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