摘要: 一、前言 默认情况下,大多数深度学习框架(比如 pytorch)都采用 32 位浮点算法进行训练。Automatic Mixed Precision(AMP, 自动混合精度)可以在神经网络训练过程中,针对不同的层,采用不同的数据精度进行计算,从而实现节省显存和加快速度的目的。 Pytorch AMP 阅读全文
posted @ 2022-11-07 16:11 小金乌会发光-Z&M 阅读(1994) 评论(0) 推荐(0) 编辑