diff --git a/config.pbtxt b/config.pbtxt new file mode 100644 index 0000000..b56e06e --- /dev/null +++ b/config.pbtxt @@ -0,0 +1,32 @@ +# Triton backend to use +name: "yolo-ref-branch" +backend: "onnxruntime" +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 + } +] +