深度学习中train()和eval()的作用
  SkyPJX5vy5h5 2023年11月02日 30 0

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

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

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

【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读
SkyPJX5vy5h5