会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
吴莫愁258
博客园
首页
新随笔
联系
订阅
管理
2021年3月11日
关于Pytorch中model.train()
摘要: model.train()将模型设置为训练状态,作用:使Dropout,batchnorm知道后有不同表现(具体参考Dropout,batchnorm源码),只有这两个关心True or False。 将模型设置为测试状态有两种方法: 1.model.train(mode=False) 2.mode
阅读全文
posted @ 2021-03-11 19:53 吴莫愁258
阅读(2269)
评论(0)
推荐(0)
编辑
公告