PyTorch 1.2.0 dirilis

PyTorch, kerangka kerja sumber terbuka populer untuk pembelajaran mesin, telah diperbarui ke versi 1.2.0. Rilis baru ini mencakup lebih dari 1900 perbaikan yang mencakup JIT, ONNX, mode pembelajaran terdistribusi, dan peningkatan kinerja.

Beberapa perubahan:

  • API TorchScript baru memungkinkan Sangat mudah untuk mengonversi nn.Module (termasuk submodul dan metode yang dipanggil di forward()) ke ScriptModule.
  • Bersama dengan Microsoft, dukungan penuh untuk ONNX Opset versi 7 (v1.2), 8 (v1.3), 9 (v1.4) dan 10 (v1.5) telah ditambahkan. Selain itu, pengguna kini dapat mendaftarkan simbol mereka sendiri untuk ekspor operasi kustom dan menentukan ukuran input dinamis selama ekspor.
  • dukungan papan tensor tidak lagi eksperimental.
  • Menambahkan modul nn.Transformer berdasarkan artikel Perhatian Yang Anda Butuhkan.
  • Banyak perbaikan pada C++ API.

Sumber: linux.org.ru

Tambah komentar