From c2917daf019cbcbf5c42ec31e1517b682fe721f8 Mon Sep 17 00:00:00 2001 From: gyeongmin Date: Tue, 13 May 2025 20:08:42 +0900 Subject: [PATCH] add config.pbtxt --- config.pbtxt | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) create mode 100644 config.pbtxt diff --git a/config.pbtxt b/config.pbtxt new file mode 100644 index 0000000..6f1553b --- /dev/null +++ b/config.pbtxt @@ -0,0 +1,26 @@ +# Triton backend to use +name: "fashion-mnist" +backend: "tensorflow_savedmodel" +max_batch_size: 0 + +# Triton should expect as input a single string +# input of variable length named 'text_input' +input [ + { + name: "Conv1_input" + data_type: TYPE_FP32 + dims: [-1, 28, 28, 1] + } +] + + +# Triton should expect to respond with a single string +# output of variable length named 'text_output' +output [ + { + name: "Softmax" + data_type: TYPE_FP32 + dims: [-1, 10] + } +] +