指定模型输入数据的shape范围。该参数预留,当前版本暂不支持。
该参数不能与--dynamic_batch_size、--dynamic_image_size、--dynamic_dims同时使用。
参数值:模型输入数据的shape范围信息,例如:"input_name1:[n1~n2,c1,h1,w1];input_name2:[n2,c2,h2,w2]"。指定的节点必须放在双引号中,节点中间使用英文分号分隔。input_name必须是转换前的网络模型中的节点名称。
参数值约束:
无。
--input_shape_range="input1:[8~20,3,5,-1];input2:[5,3~9,10,-1]"
昇腾310 AI处理器
昇腾310P AI处理器
昇腾910 AI处理器
如果模型转换时通过该参数设置了shape的范围,则使用应用工程进行模型推理时,需要在aclmdlExecute接口之前,增加aclmdlSetDatasetTensorDesc接口,用于设置真实的shape范围。
关于aclmdlSetDatasetTensorDesc接口的具体使用方法,请参见《应用软件开发指南 (C&C++)》手册“AscendCL API参考”。