Add TorchScript test file (#842)

lutzroeder · Jul 23, 2023 · 3153e0e · 3153e0e
1 parent 1bce11b
commit 3153e0e
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
@@ -2106,6 +2106,7 @@ pytorch.jit.Execution = class extends pytorch.Execution {
                                     output.resize_([ NaN, NaN, NaN, NaN, NaN ]);
                                     break;
                                 }
+                                case 'aten::roll':
                                 case 'aten::detach':
                                 case 'aten::mean':
                                 case 'aten::mul':

diff --git a/test/models.json b/test/models.json
@@ -4505,6 +4505,14 @@
     "format":   "TorchScript v1.0",
     "link":     "https://github.com/ApolloAuto/apollo"
   },
+  {
+    "type":     "pytorch",
+    "target":   "lite_real_sr.ptl",
+    "source":   "https://github.com/lutzroeder/netron/files/12137599/lite_real_sr.ptl.zip[lite_real_sr.ptl]",
+    "format":   "TorchScript v1.12",
+    "action":   "skip-render",
+    "link":     "https://github.com/lutzroeder/netron/issues/842"
+  },
   {
     "type":     "pytorch",
     "target":   "LMModel1.pt",