diff --git a/config.pbtxt b/config.pbtxt new file mode 100644 index 0000000..9e16947 --- /dev/null +++ b/config.pbtxt @@ -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 + } +] +