会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
平凡之上
博观而约取,厚积而薄发! 守得云开见月明。
博客园
首页
新随笔
新文章
联系
订阅
管理
with torch.no_grad()用处
1、可以减少内存使用
2、训练集训练好模型后,在验证集这里使用with torch.no_grad(),训练集则不会计算梯度值,然后并不会改变模型的参数,只是看了训练的效果。
posted on
2021-12-02 14:55
平凡之上
阅读(
2540
) 评论(
0
)
编辑
收藏
举报
刷新页面
返回顶部