验证模型要将Batch Size设置为1的原因

2023-12-13 20:03:12

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

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

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

综上所述，

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

文章来源:https://blog.csdn.net/Kelly_Ai_Bai/article/details/134971032
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。