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