多batch推理时,推理时间与batchSize几乎成正比,这是为什么
收藏回复举报
多batch推理时,推理时间与batchSize几乎成正比,这是为什么
t('forum.solved') 已解决
发表于2024-11-27 16:34:11
0 查看

我使用的是sample里面的 /cplusplus/level2_simple_inference/1_classification/googlenet_imagenet_multi_batch 的例子进行推理的。模型是resnet 18,输入图像是 32x32的图像。使用atc转换成了om模型    atc --model=resnet.onnx --framework=5  --output=resnet18  --soc_version=Ascend910A  --input_shape="input:128,1,32,32" 

但是 推理时间基本跟batchsize的大小成正比(如下)。这是为什么?  同意使用NVDIA显卡,增大batchsize,推理时间基本不变。

batchSize      推理时间(ms)

128                  2.145

256                  4.061

8192                  126

本帖最后由 匿名用户2024/11/27 16:37:42 编辑

我要发帖子