PyTorch 1.2.0 dikeluarkan

PyTorch, rangka kerja sumber terbuka yang popular untuk pembelajaran mesin, telah dikemas kini kepada versi 1.2.0. Keluaran baharu termasuk lebih daripada 1900 pembaikan yang meliputi JIT, ONNX, mod pembelajaran teragih dan peningkatan prestasi.

Beberapa perubahan:

  • API TorchScript baharu membolehkan Sangat mudah untuk menukar nn.Module (termasuk submodul dan kaedah yang dipanggil ke hadapan()) kepada ScriptModule.
  • Bersama-sama dengan Microsoft, sokongan penuh untuk ONNX Opset versi 7 (v1.2), 8 (v1.3), 9 (v1.4) dan 10 (v1.5) telah ditambahkan. Selain itu, pengguna kini boleh mendaftarkan simbol mereka sendiri untuk eksport operasi tersuai dan menentukan saiz input dinamik semasa eksport.
  • sokongan papan tensor tidak lagi percubaan.
  • Menambah modul nn.Transformer berdasarkan artikel Perhatian Adalah Semua yang Anda Perlukan.
  • Banyak penambahbaikan pada C++ API.

Sumber: linux.org.ru

Tambah komen