ATC动态batch模型转换问题
收藏回复举报
ATC动态batch模型转换问题
t('forum.solved') 已解决
发表于2023-10-26 15:29:07
0 查看
我有一个.pt模型,输入是数据维度是[batch,3,256,128],我将其转换成了onnx模型,也设置好了动态batch,能够成功运行,然后想用ATC工具转换成om模型,参考资料写了可以再转换过程中添加

--dynamic_batch_size="1,2,4,8"参数

但是我每次运行batch数量不固定,最大是几十,最小是1,请问参数该如何设置参数,能不能设置成一个例如1~30的范围

我要发帖子