PyTorch 1.8.1 发布,随附 PyTorch Profiler

栏目: 软件资讯 · 发布时间: 3年前

内容简介:PyTorch 1.8.1 现已发布,这是一个小的错误修复版本。具体更新内容如下: New Features 改造 torch.profiler 中的 profiling tools。torch.profiler 子模块现在可用,它利用了新发布的 kineto 库进行性能分析。详...

PyTorch 1.8.1 现已发布,这是一个小的错误修复版本。具体更新内容如下:

New Features

  • 改造 torch.profiler 中的 profiling tools。torch.profiler 子模块现在可用,它利用了新发布的 kineto 库进行性能分析。详情可查看博客
  • 为 pytorch xla 启用 autocast(#48570)。该torch.cuda.autocast软件包现在可以与 torch xla 一起使用,以提供简单的混合精度训练。

Improvements

  • 使torch.子模块导入更易于自动完成(#52339
  • 在 ONNX 中添加对torch.{isinf,any,all}的支持(#53529
  • torch.randperm的 GPU 实现中将 thrust 替换为 cub,以高提性能(#54537

Bug 修复

Misc

  • 修复torch.distributions验证检查(#53763
  • 允许更改nn.Embedding的 padding 向量(#53447
  • 在 TorchScript 中正确 de-sugar Ellipsis(#53766
  • 当组数为 24 的倍数时,停止使用 OneDNN 进行分组卷积(#54015
  • 在 {load,store}_scalar 中使用 int8_t 代替 char(#52616
  • 制作 ideep honor torch.set_num_thread(#53871
  • 修复pixel_{un}shuffle中尺寸超出范围的问题(#54178
  • 更新 kineto 以修复 libtorch 构建(#54205
  • 修复分布式 autograd CUDA 流同步的发送/接收操作(#54358

ONNX

  • 更新 ONNX 中的错误处理以避免ValueError#53548
  • 更新嵌套结构和 dict 输出的 assign output shape(#53311
  • 更新 embedding export wrt padding_idx(#53931

Documentation

  • torch.fx的文档更新(#53674
  • 修复distributed.rpc.options.TensorPipeRpcBackendOptions.set_device_map#53508
  • nn.LSTMCell的更新示例(#51983
  • 更新nn.Embeddingpadding_idx参数文档(#53809
  • 更新通用文档模板(#54141

详情可查看:https://github.com/pytorch/pytorch/releases/tag/v1.8.1


以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

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

亚马逊跨境电商运营宝典

亚马逊跨境电商运营宝典

老魏 / 电子工业出版社 / 2018-6 / 69

《亚马逊跨境电商运营宝典》详细讲解亚马逊的平台知识和运营技巧,内容围绕亚马逊卖家的日常工作,系统讲解亚马逊账号注册、后台操作、选品、产品发布、Listing 优化、站内广告、FBA 发货、VAT 税务等内容,并且通过大量的案例分析,用生动翔实的案例为读者传递运营中必备的操作技巧和运营方法。 《亚马逊跨境电商运营宝典》内容针对性强,讲解的知识、技巧和方法都充分考虑到易学、易懂、易操作、易落地执......一起来看看 《亚马逊跨境电商运营宝典》 这本书的介绍吧!

随机密码生成器
随机密码生成器

多种字符组合密码

html转js在线工具
html转js在线工具

html转js在线工具

HEX HSV 转换工具
HEX HSV 转换工具

HEX HSV 互换工具