2 changed files with 20 additions and 0 deletions
--- a/config.pbtxt
+++ b/config.pbtxt
@ -0,0 +1,20 @@
+backend: "pytorch"
+platform: "pytorch_libtorch"
+
+max_batch_size: 16
+
+input [
+  {
+    name: "input__0"
+    data_type: TYPE_FP32
+    dims: [ 3, 224, 224 ]
+  }
+]
+
+output [
+  {
+    name: "output__0"
+    data_type: TYPE_FP32
+    dims: [ 1000 ]
+  }
+]
--- a/model.pt
+++ b/model.pt