ocr_tr_engine/config.pbtxt

name:"ocr_tr_engine",
platform:"tensorrt_plan",

input [
    {
        name : "input"
        data_type : TYPE_FP32
        dims : [-1,1,32,256]
    }
]

output [
    {
        name : "output"
        data_type : TYPE_FP32
        dims : [-1,65,1698]
    }
]

model_warmup [
  {
    name : "warmup input"
    batch_size: 1
    inputs {
      key: "input"
      value: {
        data_type: TYPE_FP32
        dims: [ 1, 1, 32, 256 ]
        random_data: true
      }
    }
    count: 2
  }
]