24 lines
731 B
YAML
24 lines
731 B
YAML
model_parameters:
|
|
onnx_model: '{onnx_name_BPU_ACTPolicy_VisionEncoder}'
|
|
march: "{opt.type}"
|
|
layer_out_dump: False
|
|
working_dir: 'bpu_model_output'
|
|
output_model_file_prefix: 'BPU_VisionEncoder'
|
|
input_parameters:
|
|
input_name: ""
|
|
input_type_rt: 'featuremap'
|
|
input_layout_rt: 'NCHW'
|
|
input_type_train: 'featuremap'
|
|
input_layout_train: 'NCHW'
|
|
norm_type: 'no_preprocess'
|
|
calibration_parameters:
|
|
cal_data_dir: '{calbrate_data_name_BPU_ACTPolicy_VisionEncoder}'
|
|
cal_data_type: 'float32'
|
|
calibration_type: 'default'
|
|
optimization: set_all_nodes_int16
|
|
compiler_parameters:
|
|
extra_params: {'input_no_padding': True, 'output_no_padding': True}
|
|
jobs: 6
|
|
compile_mode: 'latency'
|
|
debug: true
|
|
optimize_level: 'O2' |