0
点赞
收藏
分享

微信扫一扫

Pytorch-Lightning混合精度训练

q松_松q 2022-01-24 阅读 96

Pytorch-Lightning混合精度训练

通用半精度训练

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

相关推荐

0 条评论