models/vamc.yaml

dataset:
  path: ILSVRC2012_img_calib
  sampler: {get_data_num: 1000, suffix: JPEG}
  transform_ops:
  - {type: DecodeImage}
  - size: [256, 256]
    type: Resize
  - crop_size: [224, 224]
    type: CenterCrop
  - mean: [0.485, 0.456, 0.406]
    std: [0.229, 0.224, 0.225]
    type: Normalize
  - {type: ToTensor}
engine:
  add_extra_ops_to_graph: {type: softmax}
  calibration: {calibrate_mode: percentile, quantize_per_channel: true, stream_mode: true}
  common: {cluster_mode: 0}
  dtype: int8
  type: vacc
model:
  checkpoint: onnx/nmv_type.onnx
  inputs:
    input: [1, 3, 224, 224]
  name: nmv_type
workspace: {enable_ir: false, save_log: true, work_dir: ./deploy_weights/}