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