test_custom_op.yml 405 字节
Newer Older
W
wangguanzhong 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
ENV:
  run_mode: paddle
  device: GPU
  min_subgraph_size: 3
  trt_calib_mode: False
  cpu_threads: 1
  trt_use_static: False
  save_img: False
  save_res: False
  return_res: False

MODEL:
  - BlankOp:
      name: blank
      param_path: paddlecv://models/PPLCNet_x1_0_infer/inference.pdiparams
      model_path: paddlecv://models/PPLCNet_x1_0_infer/inference.pdmodel
      Inputs:
        - input.image