2021年1月22日
Pytorch自动混合精度(AMP)介绍与使用
摘要： 背景： pytorch从1.6版本开始，已经内置了torch.cuda.amp，采用自动混合精度训练就不需要加载第三方NVIDIA的apex库了。本文主要从三个方面来介绍AMP： 一．什么是AMP? 二．为什么要使用AMP？ 三．如何使用AMP? 四. 注意事项 正文： 一．什么是AMP? 默认情况 阅读全文
posted @ 2021-01-22 19:51 jimchen1218 阅读(28595) 评论(1) 推荐(8)