Przeglądaj źródła

Update torch-metadata.json

Lutz Roeder 5 lat temu
rodzic
commit
956d3f8a92
1 zmienionych plików z 10 dodań i 0 usunięć
  1. 10 0
      source/torch-metadata.json

+ 10 - 0
source/torch-metadata.json

@@ -253,6 +253,16 @@
       ]
     }
   },
+  {
+    "name": "nn.SpatialMaxUnpooling",
+    "schema": {
+      "category": "Pool",
+      "attributes": [ 
+        { "name": "oheight", "visible": false },
+        { "name": "owidth", "visible": false }
+      ]
+    }
+  },
   {
     "name": "cudnn.SpatialMaxPooling",
     "schema": {