Objavljen PyTorch 1.2.0

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

Neke promjene:

  • Novi TorchScript API dozvoljava Lako je pretvoriti nn.Module (uključujući podmodule i metode pozvane u forward()) u ScriptModule.
  • Zajedno sa Microsoftom, dodana je puna podrška za ONNX Opset verzije 7 (v1.2), 8 (v1.3), 9 (v1.4) i 10 (v1.5). Osim toga, korisnici sada mogu registrirati vlastite simbole za izvoz prilagođenih operacija i specificirati dinamičke veličine unosa tokom izvoza.
  • tensorboard podrška nije više eksperimentalni.
  • Dodan modul nn.Transformer na osnovu članka Pažnja je sve što vam treba.
  • Brojna poboljšanja C++ API-ja.

izvor: linux.org.ru

Dodajte komentar