add pytorch mnist

2024-09-05 18:12:09 +09:00 · 2024-09-05 18:12:09 +09:00 · 89931b8280
commit 89931b8280
parent 095be78fdb
2 changed files with 25 additions and 0 deletions
--- a/1/model.pt
+++ b/1/model.pt
--- a/config.pbtxt
+++ b/config.pbtxt
@ -0,0 +1,22 @@
+name: "pytorch_fashionmnist"
+platform: "pytorch_libtorch"
+max_batch_size: 64
+input [
+  {
+    name: "input__0"
+    data_type: TYPE_FP32
+    dims: [ 1, 28, 28 ]
+  }
+]
+output [
+  {
+    name: "output__0"
+    data_type: TYPE_FP32
+    dims: [ 10 ]
+  }
+]
+instance_group [
+  {
+    kind: KIND_GPU
+  }
+]