压缩优化功能配置文件路径以及文件名,通过该参数使能配置文件中指定的压缩优化特性,从而提升网络性能。
参数值:配置文件路径以及文件名。
参数值格式:路径和文件名:支持大小写字母(a-z,A-Z)、数字(0-9)、下划线(_)、中划线(-)、句点(.)、中文字符。
参数值约束:
当前仅支持配置如下两种压缩方式,用户根据实际情况决定配置哪种压缩方式:
enable_first_layer_quantization:true calibration: { input_data_dir: ./data.bin,d2.bin input_shape: in:16,16;in1:16,16 config_file: simple_config.cfg infer_soc: xxxxxx infer_ip: x.x.x.x infer_port: 1000 log: info }
其中:
使能enable_first_layer_quantization特性时,只有网络结构中存在AIPP+CONV2D结构,并且在atc命令中将--enable_small_channel参数设置为1时,才有可能获得性能收益。由于量化后的模型存在一定程度上的精度损失,用户根据实际情况决定是否使能该特性。
为防止日志文件、record文件、graph文件持续落盘导致磁盘被写满,请及时清理这些文件。
参数默认值:无。
无
假设压缩优化功能配置文件名称为compression_optimize.cfg,文件内容配置示例如下:
enable_first_layer_quantization:true calibration: { input_data_dir: ./data.bin;d2.bin input_shape: in:16,16;in1:16,16 config_file: simple_config.cfg infer_soc: xxxxxxxx infer_device_id: 0 infer_ip: x.x.x.x infer_port: 1000 log: info }
将该文件上传到ATC工具所在服务器,例如上传到$HOME/module,使用示例如下:
--compression_optimize_conf=$HOME/module/compression_optimize.cfg
昇腾310B AI处理器