model-repository / catnet /config.pbtxt
jtz18's picture
Update catnet/config.pbtxt
a554cfb verified
raw
history blame contribute delete
575 Bytes
name: "catnet"
platform: "onnxruntime_onnx"
input [
{
name: "input_image"
data_type: TYPE_FP32
dims: [ 1, 24, -1, -1 ]
},
{
name: "input_label"
data_type: TYPE_INT64
dims: [ 1, -1, -1 ]
},
{
name: "input_qtable"
data_type: TYPE_FP32
dims: [ 1, 1, 8, 8 ]
}
]
output [
{
name: "out_1"
data_type: TYPE_FP32
dims: [ 1 ]
},
{
name: "output_pred"
data_type: TYPE_FP32
dims: [ -1, 2, -1, -1 ]
}
]
optimization { execution_accelerators {
cpu_execution_accelerator : [ {
name : "openvino"
} ]
}}