PyTorch 1.2.0 發布

PyTorch 是一種流行的機器學習開源框架,已更新至版本 1.2.0。 新版本包含 1900 多個修復,涵蓋 JIT、ONNX、分散式學習模式和效能改進。

一些變化:

  • 新的 TorchScript API 它允許 將 nn.Module(包括在forward() 中呼叫的子模組和方法)轉換為 ScriptModule 很容易。
  • 與 Microsoft 一起新增了對 ONNX Opset 版本 7 (v1.2)、8 (v1.3)、9 (v1.4) 和 10 (v1.5) 的全面支援。 此外,使用者現在可以註冊自己的符號以進行自訂操作匯出,並在匯出期間指定動態輸入大小。
  • 張量板支援 不再是 實驗性的。
  • 在文章的基礎上加入了 nn.Transformer 模組 注意就是您所需要的.
  • 對 C++ API 進行了大量改進。

來源: linux.org.ru

添加評論