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


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

查看所有标签

猜你喜欢:

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

JUnit Recipes中文版

JUnit Recipes中文版

陈浩等译 / 电子工业 / 2006-9 / 69.00元

《JUnit Recipes中文版:程序员实用测试技巧》主要介绍了在Java开发中使用JUnit进行单元测试的各种方法、原则、技巧与实践。本书出自开发一线专家之手,本着实用的原则,涵盖各类Java开发中应用JUnit的实用技巧,内容丰富、全面深入;无论对于需要应用JUnit进行单元测试的一线Java开发人员,还是JUnit入门、进阶者,本书都是一本不可多得的实用指南。这本书介绍了大量的JUnit实......一起来看看 《JUnit Recipes中文版》 这本书的介绍吧!

Markdown 在线编辑器
Markdown 在线编辑器

Markdown 在线编辑器

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

正则表达式在线测试

RGB CMYK 转换工具
RGB CMYK 转换工具

RGB CMYK 互转工具