--input_shape

功能说明

指定模型输入数据的shape。

关联参数

若未使用--evaluator参数，则该参数必填。

参数取值

参数值：模型输入的shape信息。

参数值格式："input_name1:n1,c1,h1,w1;input_name2:n2,c2,h2,w2"。

参数值约束：指定的节点必须放在双引号中，节点中间使用英文分号分隔。input_name必须是转换前的网络模型中的节点名称。

推荐配置及收益

无。

示例

--input_shape="input_name1:n1,c1,h1,w1;input_name2:n2,c2,h2,w2"

使用约束

• 该参数只在训练后量化场景使用。
• 如果模型的shape不固定，例如"input_name1:?,h,w,c"，其中“?”为每个batch使用的图片数量，该场景下图片数量要等于处理成bin格式数据集的图片数量。