PyTorch experiment (#133)

setup.py

@@ -93,6 +93,7 @@ def package_version():
             'caffe-model.js', 'caffe-metadata.json', 'caffe.js',
             'caffe2-model.js', 'caffe2-metadata.json', 'caffe2.js',
             'mxnet-model.js', 'mxnet-metadata.json', 'unzip.js',
+            'pytorch-model.js', 'pytorch-metadata.json',
             'numpy.js',
             'view-browser.html', 'view-browser.js',
             'view-render.css', 'view-render.js',

src/pytorch-metadata.json

@@ -0,0 +1,60 @@
+[
+    {
+        "name": "Conv2d",
+        "schema": {
+            "category": "Layer",
+            "attributes": [
+                { "name": "output_padding", "visible": false },
+                { "name": "training", "visible": false },
+                { "name": "in_channels", "visible": false },
+                { "name": "out_channels", "visible": false },
+                { "name": "groups", "visible": false }
+            ]
+        }
+    },
+    {
+        "name": "Linear",
+        "schema": {
+            "category": "Layer",
+            "attributes": [
+                { "name": "training", "visible": false }
+            ]
+        }
+    },
+    {
+        "name": "ReLU",
+        "schema": {
+            "category": "Activation",
+            "attributes": [
+                { "name": "training", "visible": false }
+            ]
+        }
+    },
+    {
+        "name": "MaxPool2d",
+        "schema": {
+            "category": "Pool",
+            "attributes": [
+                { "name": "training", "visible": false }
+            ]
+        }
+    },
+    {
+        "name": "BatchNorm2d",
+        "schema": {
+            "category": "Normalization",
+            "attributes": [
+                { "name": "training", "visible": false }
+            ]
+        }
+    },
+    {
+        "name": "Dropout",
+        "schema": {
+            "category": "Dropout",
+            "attributes": [
+                { "name": "training", "visible": false }
+            ]
+        }
+    }
+]