Objavljen PyTorch 1.2.0

PyTorch, popularni okvir otvorenog koda za strojno učenje, ažuriran je na verziju 1.2.0. Novo izdanje uključuje više od 1900 popravaka koji pokrivaju JIT, ONNX, distribuirane načine učenja i poboljšanja performansi.

Neke promjene:

  • Novi TorchScript API to omogućuje Lako je pretvoriti nn.Module (uključujući podmodule i metode pozvane u forward()) u ScriptModule.
  • Zajedno s Microsoftom dodana je puna podrška za ONNX Opset verzije 7 (v1.2), 8 (v1.3), 9 (v1.4) i 10 (v1.5). Dodatno, korisnici sada mogu registrirati vlastite simbole za prilagođene izvoze operacija i odrediti dinamičke ulazne veličine tijekom izvoza.
  • tensorboard podrška više nije eksperimentalni.
  • Dodan modul nn.Transformator na temelju članka Pažnja je sve što trebate.
  • Brojna poboljšanja C++ API-ja.

Izvor: linux.org.ru

Dodajte komentar