29 lines
1.2 KiB
YAML
29 lines
1.2 KiB
YAML
calibration_parameters:
|
|
cal_data_dir: '{dit_cal_name}/x/;{dit_cal_name}/freq/;{dit_cal_name}/t/;{dit_cal_name}/lang_c/;{dit_cal_name}/img_c/;{dit_cal_name}/lang_mask/;'
|
|
quant_config: dit_json_name
|
|
run_on_cpu: '/t_embedder/Cos;/t_embedder/Sin;/freq_embedder/Cos;/freq_embedder/Sin'
|
|
compiler_parameters:
|
|
compile_mode: latency
|
|
core_num: 1
|
|
debug: true
|
|
jobs: 8
|
|
max_time_per_fc: 0
|
|
optimize_level: O2
|
|
advice: 1
|
|
input_parameters:
|
|
input_layout_rt: NCHW;NCHW;NCHW;NCHW;NCHW;NCHW
|
|
input_layout_train: NCHW;NCHW;NCHW;NCHW;NCHW;NCHW
|
|
input_name: x;freq;t;lang_c;img_c;lang_mask;
|
|
input_shape: 1x65x1024;1;1;1x64x1024;1x4374x1024;1x64
|
|
input_space_and_range: ''
|
|
input_type_rt: featuremap;featuremap;featuremap;featuremap;featuremap;featuremap
|
|
input_type_train: featuremap;featuremap;featuremap;featuremap;featuremap;featuremap
|
|
norm_type: no_preprocess;no_preprocess;no_preprocess;no_preprocess;no_preprocess;no_preprocess
|
|
model_parameters:
|
|
layer_out_dump: false
|
|
debug_mode: "dump_calibration_data"
|
|
enable_vpu: True
|
|
march: {opt.march}
|
|
onnx_model: {dit_name}
|
|
output_model_file_prefix: rdt_dit
|
|
working_dir: bpu_output |