從實際使用的感受來說,數(shù)據(jù)及其他參數(shù)一致時,batch size設(shè)置過大,回歸類任務(wù)的mse會增大;而在分類任務(wù)中,預(yù)測精度在batch size較大時,效果幾乎無變化(待補充在batch size變化時,epoch數(shù)需要如何變化,才能保持預(yù)測精度不變)。推測batch size在一定范圍內(nèi)增大時,達(dá)到相同效果,可以減少訓(xùn)練時間,或者減少訓(xùn)練epoch數(shù)。
- 回歸任務(wù)
圖1:batch size=50時的預(yù)測圖

image.png
圖2:batch size=500時的預(yù)測圖

image.png
- 二分類預(yù)測任務(wù)
圖1:batch size=50時的預(yù)測結(jié)果及l(fā)oss曲線

image.png

屏幕快照 2019-08-28 上午11.32.37.png
圖2:batch size=500時的預(yù)測結(jié)果及l(fā)oss曲線

image.png

屏幕快照 2019-08-28 上午10.57.13.png