ATC转模型参数dynamic_batch_size设置动态BatchSize组合档数
收藏回复举报
ATC转模型参数dynamic_batch_size设置动态BatchSize组合档数
t('forum.solved') 已解决
发表于2025-02-21 11:35:38
0 查看

【需求】

最近在研究转模型设置动态batch内容,看到--dynamic_batch_size-输入选项-基础功能参数-参数说明-使用ATC工具转换模型-附录-Atlas 200I DK A2开发者套件23.0.RC3开发文档-昇腾社区有设置批次的参数,需要与参数--input_shape配合使用,但不太理解下面给的示例和档数说明。

【问题】

D20E9B0D-35D1-423f-BC8A-F91D35F348CE.png

示例--input_shape="data:-1,3,416,416;img_info:-1,4"  --dynamic_batch_size="1,2,4,8"给出的命令组合,data:-1,3,416,416,对应img_info设置-1,4,下面说“--input_shape”中的“-1”表示设置动态BatchSize,那img_info设置的-1和4分别是什么含义,该怎么理解。

给的组合档数有4组,例如data(2,3,416,416)+img_info(2,4),这里面data后面第一个数是2,和img_info设置的2和4分别指什么,该怎么理解。示例里面data后面写-1,4组档数组合写的是1,2,4,8,让人很费解。

我要发帖子