Update config.pbtxt

2025-11-12 02:15:51 +00:00 · 2025-11-12 02:15:51 +00:00 · 58a757858c
commit 58a757858c
parent b35dcc2f21
1 changed files with 10 additions and 11 deletions
--- a/config.pbtxt
+++ b/config.pbtxt
@ -1,21 +1,20 @@
-# Triton backend to use
-name: "resnet50"
 backend: "pytorch"
 platform: "pytorch_libtorch"
-max_batch_size: 0
+
+max_batch_size: 16

 input [
+  {
+    name: "input__0"
+    data_type: TYPE_FP32
+    dims: [ 3, 224, 224 ]
+  }
 ]

 output [
-]
-
-
-
-
-instance_group [
  {
-    kind: KIND_AUTO
-    count: 1
+    name: "output__0"
+    data_type: TYPE_FP32
+    dims: [ 1000 ]
  }
 ]