会员
周边
众包
新闻
博问
闪存
赞助商
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
wieneralan
博客园
首页
新随笔
联系
订阅
管理
2022年11月26日
torch中model.train() model.eval() 作用
摘要: model.train():在使用pytorch构建神经网络的时候,训练过程中会在程序上方添加一句model.train(),作用是启用batch normalization和drop out。 model.eval():测试过程中会使用model.eval(),这时神经网络会沿用batch nor
阅读全文
posted @ 2022-11-26 17:47 wieneralan
阅读(83)
评论(0)
推荐(0)
编辑
公告