会员
周边
捐助
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
阿刚的代码进阶之旅
博客园
首页
新随笔
联系
管理
订阅
2019年5月6日
pytorch使用Tips
摘要: 1、构建model的时候,写前向计算forward的时候,有时候需要判断是否是在训练的过程中,可以使用self.training来判断是否是在训练的过程中,self.train是从父类nn.Module继承而来的。 2、对于prune, running_mean和running_var的处理也是重要
阅读全文
posted @ 2019-05-06 15:34 阿刚的代码进阶之旅
阅读(197)
评论(0)
推荐(0)
编辑
pytorch使用tensorboardX
摘要: 见https://www.cnblogs.com/kk17/p/10077335.html
阅读全文
posted @ 2019-05-06 08:28 阿刚的代码进阶之旅
阅读(279)
评论(0)
推荐(0)
编辑
公告