pytorch使用Tips

1、构建model的时候,写前向计算forward的时候,有时候需要判断是否是在训练的过程中,可以使用self.training来判断是否是在训练的过程中,self.train是从父类nn.Module继承而来的。

2、对于prune, running_mean和running_var的处理也是重要的。

原文地址:https://www.cnblogs.com/yanxingang/p/10819911.html