【AI前沿动态】Pytorch 1.2.0 来了！

今天，PyTorch官方正式发布 1.2.0 版本！

此次升级有超过1900个commit，包括在JIT、ONNX、分布式、性能以及前端等方面的大量改进。

PyTorch在GitHub中对 1.2.0 版本的详细介绍：

https://github.com/pytorch/pytorch/releases

[JIT]新 TorchScript API

1.2.0 版本包含了一个新的API，它可以使 nn.Module 跟易于转换为 ScriptModule。示例如下：

[JIT]改进了TorchScript Python语言覆盖

在 1.2.0 版本中，TorchScript大大改进了对Python语言构造和Python标准库的支持。亮点包括：

提前返回(return)、中断(break)以及继续(continue)；

基于迭代器的构造，例如 for...in 循环，zip() 和 enumerate()；

NamedTuples；

支持 math 和 string 库；

支持大多数Python内置函数。

NN Transformer

错误检查：变量版本跟踪性能改进

【JIT】面向脚本模块的优化器现在是一个背景管理器。

新特性

包括张量类型支持、NN Package、操作器、Optim Package、分布式Package、可迭代数据集、Tensorboard Package等

一些不再使用的功能

支持更多的ONNX 操作集

对脚本模块的扩展支持

支持导出更多的Torch操作器到ONNX

扩展现有的导出逻辑

对导出的ONNX图进行了优化

其他一些bug修正和改进

https://github.com/pytorch/pytorch/releases

转自：新智元

EI 人工智能 AI

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。