PyTorch 1.8.1 发布,随附 PyTorch Profiler
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(#537...