(DeepLearning笔记)语义分割任务中的BatchSize

在语义分割(Semantic Segmentation)任务中,batch size的大小设置的越大越好。

例如:在使用U-Net的分割任务上,batch size设置为1反而训练结果不好。

原因:

Batch Size设置的越大,带来的直接影响是BatchNorm计算出来的均值&方差会更接近训练集数据分布的真实均值&方差,从而提高正则化的效果。

因此在计算资源允许的情况下,增大BatchSize不仅可以加快训练速度,还可以提高精度。

你可能感兴趣的:(笔记,深度学习,计算机视觉,机器学习)