Add TorchScript test file (#842) (#1001)

lutzroeder · Nov 24, 2022 · 8ebadc1 · 8ebadc1
1 parent 43ab973
commit 8ebadc1
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/source/pytorch.js b/source/pytorch.js
@@ -1978,6 +1978,7 @@ pytorch.Container.Zip.Execution = class extends pytorch.Execution {
                     for (let i = 0; i < schema.outputs.length; i++) {
                         const parameter = schema.outputs[i];
                         switch (parameter.type) {
+                            case 'Scalar':
                             case 'Tensor': {
                                 const output = this.invoke('torch.Tensor', []);
                                 output.__origin__ = type;

diff --git a/test/models.json b/test/models.json
@@ -4362,6 +4362,14 @@
     "format":   "TorchScript v1.3",
     "link":     "https://github.com/lutzroeder/netron/issues/281"
   },
+  {
+    "type":     "pytorch",
+    "target":   "mask_model.pt",
+    "source":   "https://github.com/lutzroeder/netron/files/10080302/mask_model.pt.zip[mask_model.pt]",
+    "format":   "TorchScript v1.7",
+    "error":    "Unsupported torch.add expression type in 'mask_model.pt'.",
+    "link":     "https://github.com/lutzroeder/netron/issues/842"
+  },
   {
     "type":     "pytorch",
     "target":   "mask_r_cnn.pth",