pytorch 换版本_新版 PyTorch 1.2 已发布:功能更多、兼容更全、操作更快!

PyTorch发布了1.2版本,增强了TorchScript,扩展了ONNX导出支持,新增nn.Transformer模块。更新还包括torchvision 0.4、torchaudio 0.3和torchtext 0.4,提供了更多的数据集和预处理功能。PyTorch 1.2还改进了TensorBoard的使用和对ONNX Opset的支持。
摘要由CSDN通过智能技术生成

雷锋网 AI 开发者按:近日,PyTorch 社区又添入了「新」工具,包括了更新后的 PyTorch 1.2,torchvision 0.4,torchaudio 0.3 和 torchtext 0.4。每项工具都进行了新的优化与改进,兼容性更强,使用起来也更加便捷。PyTorch 发布了相关文章介绍了每个工具的更新细节,雷锋网 AI 开发者将其整理与编译如下。

PyTorch 简介

自 PyTorch 1.0 发布以来,我们的社区不断在进行扩展、添加入新的工具。这些发展为 PyTorch Hub 中越来越多可用的模型做出了极大的贡献,并不断增加了其在研究和生产中的用途。

我们的核心目标在于通过不断向 PyTorch 中添加新功能来支持更多的研究和生产,其中也包括通过 TorchScript 来将程序与 PyTorch 连接起来。TorchScript 是一种使用 PyTorch 代码创建可序列化和可优化模型的方法;任何 TorchScript 程序都可以从 Python 进程中保存,并在不依赖于 Python 的进程中实现加载。(https://pytorch.org/docs/stable/jit.html)

而现在,我们很高兴地宣布我们已经成功开发了四个新版本,包括 PyTorch 1.2,torchvision 0.4,torchaudio 0.3 和 torchtext 0.4。用户现在可以在 pytorch.org 上(https://pytorch.org/get-started/locally/)开始使用这些版本。

PyTorch 1.2

通过使用 PyTorch 1.2,开源 ML 框架在生产应用方面向前迈出了一大步,并增加了一个改进的、更加完善的 TorchScript 环境。这些改进使得用户可以更容易建立生产模型,扩展对导出 ONNX 格式模型的支持功能,并增强对 Transformers 模块的支持。

除了这些新功能之外,用户对 TensorBoard(https://pytorch.org/docs/stable/tensorboard.html)的使用也更便捷,现在只需输入代码「from torch.utils.tensorboard import SummaryWriter」,即可开始使用。

TorchScript 改进

在 PyTorch 1.0 中加入 TorchScript 之后,它为 PyTorch 的 Eager 类模型提供了构建途径。TorchScript 编译器将 PyTorch 模型转换为静态类型的图形表示,为 Python 不可用受限环境中的优化和执行提供了机会。用户可以将模型逐步转换为 TorchScript,然后将编译后的代码与 Python 无缝混合即可。

PyTorch 1.2 还显著扩展了 TorchScript 对 PyTorch 模型中使用的 Python 子集的支持度,并提供了一种新的、更易于使用的 API,用于将模型编译为 TorchScript。有关详细信息,可以参阅迁移指南(https://pytorch.org/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值