摘要:
1、构建model的时候,写前向计算forward的时候,有时候需要判断是否是在训练的过程中,可以使用self.training来判断是否是在训练的过程中,self.train是从父类nn.Module继承而来的。 2、对于prune, running_mean和running_var的处理也是重要 阅读全文
posted @ 2019-05-06 15:34
阿刚的代码进阶之旅
阅读(207)
评论(0)
推荐(0)
摘要:
见https://www.cnblogs.com/kk17/p/10077335.html 阅读全文
posted @ 2019-05-06 08:28
阿刚的代码进阶之旅
阅读(284)
评论(0)
推荐(0)

浙公网安备 33010602011771号