Pytorch-Lightning混合精度训练

Pytorch-Lightning混合精度训练

通用半精度训练

  • Trainer构造函数中设置precision=16
  • CPU不支持半精度训练

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