PyTorch 1.5.0 发布

流行的机器学习框架 PyTorch 已更新至版本 1.5.0。 此版本包括对 API 的几项主要添加和改进,包括:

  • 以前被认为是实验性的 C++ API 终于稳定下来了。 用户现在可以轻松地将其模型从 Python API 转换为 C++ API。

  • torch.distributed.rpc 包已经稳定,提供了分布式学习的广泛功能,包括自动计算梯度和更新模型参数。

  • 更新了 torch_xla,这是一个使用 XLA 编译器来加速云 TPU 上的训练模型的包。

  • torcaudio、torchvision 和 torchtext 软件包也已更新,提供了用于开发处理音频、图形和文本数据的模型的工具。

  • 不再支持 Python 2。 所有进一步的开发将仅针对 Python 3 进行。

来源: linux.org.ru

添加评论