24 lines
764 B
YAML
24 lines
764 B
YAML
model_parameters:
|
|
onnx_model: '{onnx_name_BPU_ACTPolicy_TransformerLayers}'
|
|
march: "{opt.type}"
|
|
layer_out_dump: False
|
|
working_dir: 'bpu_model_output'
|
|
output_model_file_prefix: 'BPU_TransformerLayers'
|
|
input_parameters:
|
|
input_name: "{input_name_str}"
|
|
input_type_rt: '{input_type_str}'
|
|
input_layout_rt: '{nchw_str}'
|
|
input_type_train: '{input_type_str}'
|
|
input_layout_train: '{nchw_str}'
|
|
norm_type: '{norm_type_str}'
|
|
calibration_parameters:
|
|
cal_data_dir: '{cal_data_dir_str}'
|
|
cal_data_type: '{cal_data_type_str}'
|
|
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: False
|
|
optimize_level: 'O2' |