验证模型要将Batch Size设置为1的原因
·
一般为了充分地利用计算机资源,在训练的时候会将batch_size设置得很大
但是在验证模型的时候有些将Batch Size设置为1
如果要将多张图片组成一个batch送入网络进行训练,是需要将图片的大小进行统一的,只有相同尺寸的图片才能组成batch,所以如果大小不一致需要进行pad操作,但是pad操作可能会对模型的预测结果产生或多或少的影响
综上所述,
- 若输入模型数据shape固定,验证时对Batch Size无限制(比如分类网络)
- 若输入模型数据shape不固定,验证时将Batch Size设置为1
更多推荐



所有评论(0)