groupuser/resnet50
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update config.pbtxt

Browse Source
This commit is contained in:
groupuser 2025-11-12 02:15:51 +00:00
parent b35dcc2f21
commit 58a757858c
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
config.pbtxt
View File

@ -1,21 +1,20 @@
# Triton backend to use
name: "resnet50"
backend: "pytorch" backend: "pytorch"
platform: "pytorch_libtorch" platform: "pytorch_libtorch"
max_batch_size: 0
max_batch_size: 16
input [ input [
{
name: "input__0"
data_type: TYPE_FP32
dims: [ 3, 224, 224 ]
}
] ]
output [ output [
]
instance_group [
{ {
kind: KIND_AUTO name: "output__0"
count: 1 data_type: TYPE_FP32
dims: [ 1000 ]
} }
] ]