diff --git a/config.pbtxt b/config.pbtxt new file mode 100644 index 0000000..df1162b --- /dev/null +++ b/config.pbtxt @@ -0,0 +1,21 @@ +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 ] + } +] +