PyTorch 2.0正式版发布!一行代码提速2倍,100%向后兼容(1/7)

更多
2023-3-26 16:57 162人参与 0条评论 自动播放 开灯

PyTorch 2.0正式版发布!一行代码提速2倍,100%向后兼容

运用自定义内核和 torch.compile来提供对大型言语模型锻炼显著加速HuggingFace Transformers的主要维护者Sylvain Gugger在PyTorch项目发表的一份声明中写道「只需添加一行代码,PyTorch 2.0就能在锻炼Transformers模型时提供1.5倍至2.0倍的速度。这是自混合精度锻炼推出以来最令人兴奋的事情!」PyTorch和谷歌的TensorFlow是两个最盛行的深度学习框架。世界上有数千家机构正在运用PyTorch ... [查看原文]

上传图片 点击发表评论 (162人参与 0条评论)