--data_dir
功能说明
与模型匹配的bin格式数据集路径。
关联参数
若未使用--evaluator参数,则该参数必填。
参数取值
参数值:bin格式数据集路径。
参数值约束:
- 路径支持大小写字母(a-z,A-Z)、数字(0-9)、下划线(_)、中划线(-)、句点(.)、中文字符。
- 若模型有多个输入,且每个输入有多个batch数据,则不同的输入数据必须存储在不同的目录中,目录中文件的名称必须按照升序排序。所有的输入数据路径必须放在双引号中,节点中间使用英文分号分隔。
- 单个bin文件中存储的数组shape需要和--input_shape中输入的shape相匹配,例如:单张图片bin存储的数组shape为1x224x224x3,则--input_shape中输入的必须为1x224x224x3;如需多个bin做量化,则可通过调整--batch_num取值实现。
推荐配置及收益
无。
示例
--data_dir="data/input1/;data/input2/"
input1和input2目录中,数据文件名举例为input1batch1.bin、input1batch2.bin;input2batch1.bin、input2batch2.bin,文件名称按照升序进行排列。
使用约束
该参数只在训练后量化场景使用。
父主题: 可选参数