Add PyTorch PixelShuffle (#133)

lutzroeder · Nov 23, 2018 · eb99b24 · eb99b24
1 parent ea979d3
commit eb99b24
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/src/pytorch-metadata.json b/src/pytorch-metadata.json
@@ -156,13 +156,6 @@
       "package": "torch.nn.modules.activation"
     }
   },
-  {
-    "name": "Threshold",
-    "schema": {
-      "category": "Activation",
-      "package": "torch.nn.modules.activation"
-    }
-  },
   {
     "name": "ReLU",
     "schema": {
@@ -529,6 +522,12 @@
       "package": "torch.nn.modules.padding"
     }
   },
+  {
+    "name": "PixelShuffle",
+    "schema": {
+      "package": "torch.nn.modules.pixelshuffle"
+    }
+  },
   {
     "name": "InstanceNorm1d",
     "schema": {

diff --git a/src/pytorch.js b/src/pytorch.js
@@ -120,6 +120,7 @@ pytorch.ModelFactory = class {
             constructorTable['torch.nn.modules.padding.ConstantPad1d'] = function () {};
             constructorTable['torch.nn.modules.padding.ConstantPad2d'] = function () {};
             constructorTable['torch.nn.modules.padding.ConstantPad3d'] = function () {};
+            constructorTable['torch.nn.modules.pixelshuffle.PixelShuffle'] = function () {};
             constructorTable['torch.nn.modules.pooling.AvgPool1d'] = function () {};
             constructorTable['torch.nn.modules.pooling.AvgPool2d'] = function () {};
             constructorTable['torch.nn.modules.pooling.AvgPool3d'] = function () {};
@@ -138,6 +139,7 @@ pytorch.ModelFactory = class {
             constructorTable['torch.nn.modules.upsampling.Upsample'] = function() {};
             constructorTable['torch.nn.parallel.data_parallel.DataParallel'] = function() {}; 
             constructorTable['torch.nn.parameter.Parameter'] = function(data, requires_grad) { this.data = data; this.requires_grad = requires_grad; };
+            constructorTable['torch.nn.utils.spectral_norm.SpectralNorm'] = function () {};
             constructorTable['torch.nn.utils.weight_norm.WeightNorm'] = function () {};
             constructorTable['torchvision.models.alexnet.AlexNet'] = function () {};
             constructorTable['torchvision.models.densenet.DenseNet'] = function () {};