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

添加評論