0
点赞
收藏
分享

微信扫一扫

深度学习中train()和eval()的作用

(1)在训练时要加上train()。在测试和使用时要加上eval()。它们两个函数用来控制【batchNorm层,dropout层等用于优化训练而添加的网络层】的开关。

Train()会开启那些层,eval会关闭那些层。

(2)dropout层的作用,在每一次的训练过程中,随机舍弃一定比例的神经元,每次训练舍弃的神经元都不同,可以有效防止模型在训练过程中过拟合。

举报

相关推荐

0 条评论