Add TorchScript test files (#281)

lutzroeder · Jun 18, 2019 · f40fdc3 · f40fdc3
1 parent 3058053
commit f40fdc3
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/test/models.json b/test/models.json
@@ -4604,5 +4604,26 @@
     "script": [ "${root}/tools/pytorch", "sync install zoo" ],
     "format": "TorchScript v1",
     "link":   "https://pytorch.org/docs/stable/torchvision/models.html"
+  },
+  {
+    "type":   "torchscript",
+    "target": "traced_online_lane_enc.pt",
+    "source": "https://github.com/ApolloAuto/apollo/blob/master/modules/prediction/data/traced_online_lane_enc.pt?raw=true",
+    "format": "TorchScript v1",
+    "link":   "https://github.com/ApolloAuto/apollo"
+  },
+  {
+    "type":   "torchscript",
+    "target": "traced_online_obs_enc.pt",
+    "source": "https://github.com/ApolloAuto/apollo/blob/master/modules/prediction/data/traced_online_obs_enc.pt?raw=true",
+    "format": "TorchScript v1",
+    "link":   "https://github.com/ApolloAuto/apollo"
+  },
+  {
+    "type":   "torchscript",
+    "target": "traced_online_pred_layer.pt",
+    "source": "https://github.com/ApolloAuto/apollo/blob/master/modules/prediction/data/traced_online_pred_layer.pt?raw=true",
+    "format": "TorchScript v1",
+    "link":   "https://github.com/ApolloAuto/apollo"
   }
 ]