From 1d951bc9caa2c10fa339091cb997b30e83db5c16 Mon Sep 17 00:00:00 2001 From: groupuser Date: Tue, 31 Dec 2024 03:51:07 +0000 Subject: [PATCH] Upload config.pbtxt --- config.pbtxt | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 config.pbtxt diff --git a/config.pbtxt b/config.pbtxt new file mode 100644 index 0000000..d6f4eb2 --- /dev/null +++ b/config.pbtxt @@ -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 ] + } +]