Add some fields

Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com>
2020-05-30 18:00:22 +02:00
parent 64d22c51f1
commit 041a5cf65d
2 changed files with 10 additions and 0 deletions
@@ -59,6 +59,14 @@ namespace tk { namespace dnn {
            fields.batch_normalize = std::stoi(value);
        else if (name == "filters")
            fields.filters = std::stoi(value);
+        else if (name == "size")
+            fields.size = std::stoi(value);
+        else if (name == "stride")
+            fields.stride = std::stoi(value);
+        else if (name == "pad")
+            fields.pad = std::stoi(value);
+        else if (name == "activation")
+            fields.activation = value;
        
        return true;
    }