config.pbtxt 872 字节
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
name: "cls_pp"
platform: "ensemble"
max_batch_size: 128
input [
  {
    name: "x"
    data_type: TYPE_FP32
    dims: [ 3, -1, -1 ]
  }
]
output [
  {
    name: "cls_labels"
    data_type: TYPE_INT32
    dims: [ 1 ]
  },
  {
    name: "cls_scores"
    data_type: TYPE_FP32
    dims: [ 1 ]
  }
]
ensemble_scheduling {
  step [
    {
      model_name: "cls_runtime"
      model_version: 1
      input_map {
        key: "x"
        value: "x"
      }
      output_map {
        key: "softmax_0.tmp_0"
        value: "infer_output"
      }
    },
    {
      model_name: "cls_postprocess"
      model_version: 1
      input_map {
        key: "POST_INPUT_0"
        value: "infer_output"
      }
      output_map {
        key: "POST_OUTPUT_0"
        value: "cls_labels"
      }
      output_map {
        key: "POST_OUTPUT_1"
        value: "cls_scores"
      }
    }
  ]
}