在新版本中,Facebook 与 AWS 还合作共同推出了大规模生产级工具库 TorchServe。
今天,PyTorch 1.5 宣布上线,此版本主要包括几个新的 API 的添加和改进。新版 PyTorch 包括对 C++前端的重大更新,用于计算机视觉模型的「channels last」存储格式,以及用于模型并行训练的分布式 RPC 框架的稳定版本。该版本还提供了针对自动求导机制中黑塞和雅可比的新 API,以及受 pybind 启发,允许用户创建自定义 C++类的一个 API。另外,torch_xla 已可在 PyTorch 1.5 版中使用,并在 1.5 版本中进行了测试,可提供成熟的 Cloud TPU 体验。
版本说明:https://github.com/pytorch/pytorch/releases/tag/v1.5.0
主要变化
以下是 PyTorch 1.5 版本的主要变化
C++ 前端 API(稳定型)
现在 C++前端 API 与 Python 版同等丰富,之前实验性的功能都已移到「稳定版」中。主要亮点如下:
专为计算机视觉设计的「Channels last」储存格式(实验型)