一般为了充分地利用计算机资源,在训练的时候会将batch_size设置得很大

但是在验证模型的时候有些将Batch Size设置为1

如果要将多张图片组成一个batch送入网络进行训练,是需要将图片的大小进行统一的,只有相同尺寸的图片才能组成batch,所以如果大小不一致需要进行pad操作,但是pad操作可能会对模型的预测结果产生或多或少的影响

综上所述,

  • 若输入模型数据shape固定,验证时对Batch Size无限制(比如分类网络)
  • 若输入模型数据shape不固定,验证时将Batch Size设置为1
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐