diff --git a/test.txt b/test.txt new file mode 100644 index 0000000..138b0e6 --- /dev/null +++ b/test.txt @@ -0,0 +1,17 @@ +name: "pytorch-fashion-minst" +platform: "pytorch_libtorch" +max_batch_size : 0 +input [ + { + name: "input__0" + data_type: TYPE_FP32 + dims: [ 1, 1, 28, 28 ] + } +] +output [ + { + name: "output__0" + data_type: TYPE_FP32 + dims: [ 10 ] + } +] \ No newline at end of file