PyTorch 1.2.0 发布,新增TorchScript API,扩展ONNX导出

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

内容简介:PyTorch 1.2.0版正式发布,和 1.1 版本相比,新版本主要新增/改动的功能包括: 完善TorchScript环境,提供了新的将模型编译为TorchScrip的API 扩展ONNX模型的导出的支持 增强对Transformer架构的模块级支持 此外,...

PyTorch 1.2.0版正式发布,和 1.1 版本相比,新版本主要新增/改动的功能包括:

  • 完善TorchScript环境,提供了新的将模型编译为TorchScrip的API
  • 扩展ONNX模型的导出的支持
  • 增强对Transformer架构的模块级支持
  • 此外,TensorBoard现在也能快速开始使用了

PyTorch 是一个 Torch7 团队开源的 Python 优先的深度学习框架,提供两个高级功能:

  • 强大的 GPU 加速 Tensor 计算(类似 numpy)

  • 构建基于 tape 的自动升级系统上的深度神经网络

你可以重用你喜欢的 python 包,如 numpy、scipy 和 Cython ,在需要时扩展 PyTorch。


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

查看所有标签

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

美团机器学习实践

美团机器学习实践

美团算法团队 / 人民邮电出版社 / 2018-8-1 / 79.00元

人工智能技术正以一种超快的速度深刻地改变着我们的生活,引导了第四次工业革命。美团作为国内O2O领域领 先的服务平台,结合自身的业务场景和数据,积极进行了人工智能领域的应用探索。在美团的搜索、推荐、计算广告、风控、图像处理等领域,相关的人工智能技术得到广泛的应用。本书包括通用流程、数据挖掘、搜索和推荐、计算广告、深度学习以及算法工程6大部分内容,全面介绍了美团在多个重要方面对机器学习的应用。 ......一起来看看 《美团机器学习实践》 这本书的介绍吧!

在线进制转换器
在线进制转换器

各进制数互转换器

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

Markdown 在线编辑器

RGB HSV 转换
RGB HSV 转换

RGB HSV 互转工具