流行的机器学习框架 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