流行的機器學習框架 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