1,097
社区成员
发帖
与我相关
我的任务
分享
想问一下,SNPE支持dynamic shape吗,我希望像tensorrt那样,在运行的时候指定模型尺寸,而不是在尝试不同尺寸的输入时都需要更换模型,但是snpe转模型的时候我没有找到类似的指定min/max的接口,我尝试转了一个dynamic的模型,snpe报错Dynamic input batch_size not supported. Use --input_dim command to provide a static batch value。
您好,支持的。snpe在转模型的时候要求模型是static的,但是在您build SNPE实例的时候snpe允许重新推导模型尺寸。您可以在build的时候指定setInputDimensions,它的输入是zdl::DlSystem::TensorShapeMap,在这里可以指定各输入的shape。