groupuser/yolo-ref-branch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update config.pbtxt

Browse Source
This commit is contained in:
cheetahadmin 2025-09-15 06:31:48 +00:00
parent 0b2763c39b
commit 0b27d9cbda
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
config.pbtxt
View File

@ -6,20 +6,27 @@ max_batch_size: 0
# Triton should expect as input a single string # Triton should expect as input a single string
# input of variable length named 'text_input' # input of variable length named 'text_input'
input [ input [
{
name: "images"
data_type: TYPE_FP32
dims: [1, 3, 640, 640]
}
] ]
# Triton should expect to respond with a single string # Triton should expect to respond with a single string
# output of variable length named 'text_output' # output of variable length named 'text_output'
output [ output [
{
name: "output0"
data_type: TYPE_FP32
dims: [1, 84, 8400]
}
] ]
instance_group [ instance_group [
{ {
kind: KIND_AUTO kind: KIND_AUTO
count: 1 count: 1
} }
] ]