9 - nn.Module、nn.Sequential以及其他容器Module的源码讲解

文章目录

  • 1. train 训练模式
  • 2. eval 推理模式
  • 3. requires_grad_
  • 4. zero_grad
  • 5. str(mymodel)
  • 6. nn.Sequential
  • 7. nn.ModuleList
  • 8. nn.ModuleDict
  • 9. nn.ParameterList
  • 10. nn.ParameterDict

1. train 训练模式

  • 将模型设置成训练模式,本质上式将self.training=True
  • 启用dropout类和batchnorm类
    def train(self:

你可能感兴趣的:(python,pytorch,深度学习,pytorch)