如何选择minibatch size

首先,当minibatch偏小时,产生的noise能够部分避免local minima,所以相比于large minibatch, 我们偏向于使用small minibatch。
如何选择minibatch size_第1张图片
我们通常的选择是32~256之间,large minibatch(512+)计算成本高,而small minibatch(16-)训练速度太慢。
如何选择minibatch size_第2张图片

如何选择minibatch size_第3张图片
如何选择minibatch size_第4张图片

你可能感兴趣的:(机器学习,算法)