--calibration_shape

功能说明

指定模型校准输入数据的shape。

关联参数

该参数需要与--calibration_data配合使用。

参数取值

参数值:

模型输入校准数据的shape信息,例如:"input_name1:n1,c1,h1,w1;input_name2:n2,c2,h2,w2"。指定的节点必须放在双引号中,节点中间使用英文分号分隔。input_name必须是转换前的网络模型中的节点名称。注意:校准数据shape的各维度大小,即样例中的n1,c1,h1,w1等参数,不支持设置为负数。

推荐配置及收益

无。

示例

--calibration_shape="input_name1:n1,c1,h1,w1;input_name2:n2,c2,h2,w2"  --calibration_data=./tf_fcn/keep_probability.bin,./tf_fcn/tmp_featuremap_input.bin

依赖约束

无。