神经网络训练经验

【静态经验】

1. ResNet-50 3D,#param大约30+M,kinetics-400,dropout使用0.2,weight decay使用5e-4,momentum 0.9。

2. ResNet-23 2D,#param大约11M,kinetics-400,dropout使用0.5,weight decay使用1e-4, momentum0.9。

【Learning Rate】

与step调整lr相比,使用退火方式,使得训练过程更加顺滑,同时能够收敛到比较好的结果,同时更加稳定。

【Batch Size】

BN对batch size敏感,如果使用bn,则更大的batch size有利于bn更好拟合样本总体分布。

【Weight Decay】

根据目前的经验,wd和网络参数量、训练数据量有关。数据量差不多时,大网络使用大的weight decay(ResNet-50 3D,#param大约30M,kinetics-400,wd使用5e-4),小网络使用小的wd(ResNet-23 2D,#param大约11M,kinetics-400,wd使用1e-4)。

【Dropout】

根据目前的经验,小网络的do应该大,大网络的do应该小。例如ResNet-50 3D,#param大约30M,kinetics-400,do使用0.2;ResNet-23 2D,#param大约11M,kinetics-400,do使用0.5

猜你喜欢

转载自www.cnblogs.com/hizhaolei/p/10113026.html
今日推荐