PyTorch 1.1.0 发布

栏目: Python · 发布时间: 6年前

内容简介:PyTorch 1.1.0 发布了,此版本主要提高了性能、添加了新的模型理解和可视化工具以提高可用性,并提供新的 API。需要注意的是,此版本不再支持 CUDA 8.0。TensorBoard 是一个用于检查和理解训练脚本、张量和图的 Web 应用程序套件,使用它可以进行一级与原生可视化和模型调试。PyTorch 现在通过一个简单的“from torch.utils.tensorboard import SummaryWriter”命令原生支持 TensorBoard。

PyTorch 1.1.0 发布了,此版本主要提高了性能、添加了新的模型理解和可视化 工具 以提高可用性,并提供新的 API。

需要注意的是,此版本不再支持 CUDA 8.0。

TensorBoard 

TensorBoard 是一个用于检查和理解训练脚本、张量和图的 Web 应用程序套件,使用它可以进行一级与原生可视化和模型调试。PyTorch 现在通过一个简单的“from torch.utils.tensorboard import SummaryWriter”命令原生支持 TensorBoard。

PyTorch 1.1.0 发布

JIT 编译器

即时(JIT)编译有一些改进,这些改进包括各种 bug 修复与 TorchScript 中的扩展功能,例如对字典、用户类和属性的支持。

@torch.jit.script
class Pair:
	def __init__(self, first, second)
		self.first = first
		self.second = second

	def sum(self):
		return self.first + self.second

新 API

支持布尔张量、更好地支持自定义递归神经网络。

分布式训练

改进了 CNN 等常见模型的性能,增加了对多设备模块的支持,包括在使用分布式数据并行(DDP,Distributed Data Parallel)的同时跨 GPU 分割模型的能力,并支持在每次迭代中不使用所有参数的模块(例如控制流程,如自适应 softmax 等)。

关于 性能的提升 ,有以下几项数据:

nn.BatchNorm
nn.AdaptiveAvgPool
nn.EmbeddingBag
Tensor.copy_
torch.nonzero
reduction functions

详细更新内容查看:


以上所述就是小编给大家介绍的《PyTorch 1.1.0 发布》,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对 码农网 的支持!

查看所有标签

猜你喜欢:

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

京东技术解密

京东技术解密

京东研发体系 / 电子工业出版社 / 2014-11-18 / 65

京东高速的增长、闪电响应的供应链、庞大的团队规模等背后内幕,对于业界一直像谜一样神秘。随着成为中国B2C领导厂商以及在纳斯达克上市,京东越来越需要开放自己,与业界形成更好的交流与融合。《京东技术解密》的面世,就是京东技术团队首次向业界集体亮相。本书用翔实的内容为读者逐一解答——如何用技术支撑网站的综合竞争实力,如何把握技术革新的时间点,如何应对各种棘手问题及压力,如何在网站高速运转的情况下进行系统......一起来看看 《京东技术解密》 这本书的介绍吧!

CSS 压缩/解压工具
CSS 压缩/解压工具

在线压缩/解压 CSS 代码

UNIX 时间戳转换
UNIX 时间戳转换

UNIX 时间戳转换

正则表达式在线测试
正则表达式在线测试

正则表达式在线测试