yolo-ref-branch/config.pbtxt

38 lines
349 B
Plaintext

# Triton backend to use
name: "yolo-ref-branch"
platform: "onnxruntime_onnx"
max_batch_size: 0
input [
{
name: "images"
data_type: TYPE_FP32
dims: [ 1, 3, 640, 640 ]
}
]
output [
{
name: "output0"
data_type: TYPE_FP32
dims: [ 1, 84, 8400 ]
}
]
instance_group [
{
kind: KIND_AUTO
count: 1
}
]