专栏首页机器学习爱好者社区【留言送书】PyTorch1.8版本发布!更有TorchVision等期待的更新!

【留言送书】PyTorch1.8版本发布!更有TorchVision等期待的更新!

近日,Facebook发布了PyTorch 1.8新版本。自1.7版以来,此版本包含3,000多次提交。它包括众多更新和优化:编译,代码优化,用于科学计算的前端API,以及通过pytorch.org提供的二进制文件对AMD ROCm的支持。它还为管道和模型并行性以及梯度压缩的大规模训练提供了改进的功能。几个重大更新包括:

  • torch.fx:一个可以用于操作 nn.Module实例的工具,它可以改变其中的operations,比如新增一个op, FX包含三个重要部分:a symbolic tracer, an intermediate representation, and Python code generation。更多内容可以阅读文档:https://pytorch.org/docs/stable/fx.html
  • torch.fft:支持FFTs,torch.linalg:支持和numpy类似的线性代数库;
  • 分布式训练优化:Improved NCCL reliability; Pipeline parallelism support; RPC profiling; and support for communication hooks adding gradient compression.

更多更新详情见:https://pytorch.org/blog/pytorch-1.8-released/,目前官方1.8版本支持cuda10.2和cuda11.1两个版本,默认是cuda10.2:

伴随着PyTorch1.8版本的更新,同步更新的还有其它Torch库:

  • TorchVision - Added support for PyTorch Mobile including Detectron2Go (D2Go), auto-augmentation of data during training, on the fly type conversion, and AMP autocasting.
  • TorchAudio - Major improvements to I/O, including defaulting to sox_io backend and file-like object support. Added Kaldi Pitch feature and support for CMake based build allowing TorchAudio to better support no-Python environments.
  • TorchText - Updated the dataset loading API to be compatible with standard PyTorch data loading utilities.
  • TorchCSPRNG - Support for cryptographically secure pseudorandom number generators for PyTorch is now stable with new APIs for AES128 ECB/CTR and CUDA support on Windows.

其中torchvision最新版本为0.9.0,新增了MobileNetV3 模型,支持数据增强

AutoAugment,支持on-the-fly image type conversions,还有最重要的更新是对移动端的支持,重点发布了Detectron2Go (D2Go)库,其支持目标检测库落地到移动端,比如发布到安卓app上:

https://github.com/pytorch/android-demo-app/tree/master/D2Go

D2GO库继承了detectron2的风格,但是更多的是推动目标检测在移动端的落地,目前的D2GO的backbone是FB提出的fbnet,在D2GO项目中也给出了训练DETR和Deformable DETR,更多见github仓库:

https://github.com/facebookresearch/d2go/tree/master/projects_oss/detr

[ 完 ]

本文分享自微信公众号 - 机器学习爱好者社区(ML_shequ)

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2021-03-06

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • 「炼丹」师的福音!支持AMD GPU,PyTorch 1.8来了!

    近日,Facebook发布了PyTorch 1.8新版本,加入了对AMD ROCm的支持,可以不用去配置Docker在原生环境下运行。

    新智元
  • OpenCV4.4刚刚发布!支持YOLOv4、EfficientDet检测模型,SIFT移至主库!

    OpenCV 4.4.0 于2020年7月18日正式发布,不得不说OpenCV 作为最大开源的图像处理工具,提供的内容太全面了,对小白友好度很高。不仅算法众多,...

    Amusi
  • 顺藤摸瓜,突然有点心疼尤雨溪,写vue3太难了

    就在前天凌晨6点多,我在公众号上推送了一篇文 《尤雨溪:Vue3.0全球发布会在线视频》,没看过的戳标题回顾,真的建议有空多看几遍,重温下尤大帅气的脸庞,才是最...

    闰土大叔
  • 2020年10月GitHub上最热门的Python开源项目

    10月份GitHub上最热门的Python开源项目排行已经出炉啦,一起来看看上榜详情吧:

    龙哥
  • PyTorch 1.9发布,这些更新值得关注!

    不久之前,PyTorch 官方博客发布 1.8 版本,此版本由 1.7 发布以来的 3000 多个 commit 组成,重要更新内容包括在编译、代码优化和科学计...

    昱良
  • 新版 PyTorch 1.2 已发布:功能更多、兼容更全、操作更快!

    近日,PyTorch 社区又添入了「新」工具,包括了更新后的 PyTorch 1.2,torchvision 0.4,torchaudio 0.3 和 torc...

    AI研习社
  • 开发 | PyTorch好助手:PyTorch Hub一键复现各路模型

    无论 ResNet、BERT、GPT、VGG、PGAN,还是 MobileNet,只需一行代码轻松复现!

    AI科技评论
  • 6位大咖,12本好书,TVP读书分享闭门会带你开启心灵奇旅 | 内有福利放送

    古语有云,君子不学,不成其德。阅读,不仅可以充实知识,更能开拓视野,提升格局。无论何时,阅读都应是生活的刚需。新春前夕,TVP将带来“鼠实不易,牛转未来” 新年...

    TVP官方团队
  • Python猫荐书系统之四:《Python源码剖析》

    先来看看今天的主角是谁:《Python源码剖析——深度探索动态语言核心技术》,2008年出版,作者 @陈儒 ,评分8.7分。

    Python猫

扫码关注云+社区

领取腾讯云代金券