Automatically created by CHEETAH
This commit is contained in:
parent
d86e669aea
commit
be12473509
43
config.pbtxt
Normal file
43
config.pbtxt
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
# Triton backend to use
|
||||||
|
name: "yolo-ref-branch"
|
||||||
|
platform: "onnxruntime_onnx"
|
||||||
|
max_batch_size: 0
|
||||||
|
|
||||||
|
input [
|
||||||
|
|
||||||
|
{
|
||||||
|
name: "input0"
|
||||||
|
data_type: "TYPE_FP32"
|
||||||
|
dims: 1, 3, 640, 640
|
||||||
|
|
||||||
|
reshape {
|
||||||
|
shape: 255, 255, 255
|
||||||
|
}
|
||||||
|
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "input1"
|
||||||
|
data_type: "TYPE_FP32"
|
||||||
|
dims: 2, 2, 2
|
||||||
|
|
||||||
|
reshape {
|
||||||
|
shape: 8,8,8
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
]
|
||||||
|
|
||||||
|
output [
|
||||||
|
]
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
instance_group [
|
||||||
|
{
|
||||||
|
kind: KIND_AUTO
|
||||||
|
count: 1
|
||||||
|
}
|
||||||
|
]
|
||||||
|
|
||||||
Loading…
Reference in New Issue
Block a user