最新资讯新版 PyTorch 12 已发布功能丰富兼容性强操作效率提升

最新资讯:PyTorch 社区再添新工具!PyTorch 1.2、torchvision 0.4、torchaudio 0.3 和 torchtext 0.4 正式发布,带来更强的兼容性和便捷操作。这些更新为 PyTorch Hub 中越来越多可用的模型提供了极大的贡献,并增强了其在研究和生产中的应用。核心目标是通过不断向 PyTorch 中添加新功能支持更多研究和生产,其中 TorchScript 方法使得程序与 PyTorch 连接起来。

现在,我们高兴宣布开发了四个新版本,用户可以在 pytorch.org 上开始使用。PyTorch 1.2 在生产应用中迈出重要步伐,并增加了改进的 TorchScript 环境,使用户更容易建立生产模型并扩展对导出 ONNX 格式模型的支持,同时增强对 Transformers 模块的支持。

除了这些新功能外,TensorBoard 的使用也更加便捷,只需输入代码「from torch.utils.tensorboard import SummaryWriter」,即可开始使用。

关于 TorchScript 的改进,它为 Eager 类型模型提供构建途径,并且扩展了对 Python 子集的支持度,以及提供了一种新的 API 用于将模型编译为 TorchScript。此外,还有 ONNX EXPORT 改进,包括全面支持导出 ONNX Opset 版本7至10,以及 ScriptModule 的改进,如多输出、张量工厂表等。

最后,在 nn.Transformer 模块方面,我们包含一个标准模块,该模块基于 Attention is All You Need 论文,可以独立或作为组件单独使用。此外,有关完整发版说明,请参见 GitHub releases 页面。

猜你喜欢