PyTorch 1.2.0 发布，新增TorchScript API，扩展ONNX导出

内容简介：PyTorch 1.2.0版正式发布，和 1.1 版本相比，新版本主要新增/改动的功能包括： 完善TorchScript环境，提供了新的将模型编译为TorchScrip的API 扩展ONNX模型的导出的支持 增强对Transformer架构的模块级支持 此外，...

PyTorch 1.2.0版正式发布，和 1.1 版本相比，新版本主要新增/改动的功能包括：

• 完善TorchScript环境，提供了新的将模型编译为TorchScrip的API
• 扩展ONNX模型的导出的支持
• 增强对Transformer架构的模块级支持
• 此外，TensorBoard现在也能快速开始使用了

PyTorch 是一个 Torch7 团队开源的 Python 优先的深度学习框架，提供两个高级功能：

• 强大的 GPU 加速 Tensor 计算（类似 numpy）

• 构建基于 tape 的自动升级系统上的深度神经网络

你可以重用你喜欢的 python 包，如 numpy、scipy 和 Cython ，在需要时扩展 PyTorch。

以上所述就是小编给大家介绍的《PyTorch 1.2.0 发布，新增TorchScript API，扩展ONNX导出》，希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对 码农网 的支持！

查看所有标签

本站部分资源来源于网络，本站转载出于传递更多信息之目的，版权归原作者或者来源机构所有，如转载稿涉及版权问题，请联系我们。