Fork me on GitHub
摘要: 为了帮助提高Pytorch的训练效率,英伟达提供了混合精度训练工具Apex。号称能够在不降低性能的情况下,将模型训练的速度提升2-4倍,训练显存消耗减少为之前的一半。该项目开源于:https://github.com/NVIDIA/apex ,文档地址是:https://nvidia.github. 阅读全文
posted @ 2021-10-25 21:34 sy- 阅读(839) 评论(0) 推荐(0)