NVIDIA/apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

语言：Python · GitHub 星标：9,000 · Fork：1,531

在 GitHub 上查看原仓库