文件名称：PyTorch扩展：用于在Pytorch中轻松实现混合精度和分布式培训的工具-Python开发
文件大小：759KB
文件格式：ZIP
更新时间：2024-06-14 05:13:53
Python Deep Learning
简介此存储库包含NVIDIA维护的实用程序，以简化Pytorch中的混合精度和分布式培训。 此处的某些代码最终将包含在上游Pytorch中。 Apex的目的是介绍该存储库包含NVIDIA维护的实用程序，以简化Pytorch中的混合精度和分布式培训。 此处的某些代码最终将包含在上游Pytorch中。 Apex的目的是使用户可以尽快使用最新的实用程序。 完整的API文档：https://nvidia.github.io/apex GTC 2019和Pytorch DevCon 2019幻灯片内容1. Amp：自动混合精度apex.amp是启用混合精度训练的工具