PyTorch 1.2.0 utgitt

PyTorch, et populært rammeverk med åpen kildekode for maskinlæring, har blitt oppdatert til versjon 1.2.0. Den nye utgivelsen inkluderer mer enn 1900 rettelser som dekker JIT, ONNX, distribuerte læringsmoduser og ytelsesforbedringer.

Noen endringer:

  • Ny TorchScript API den lar Det er enkelt å konvertere nn.Module (inkludert undermoduler og metoder kalt forward()) til ScriptModule.
  • Sammen med Microsoft er full støtte for ONNX Opset versjon 7 (v1.2), 8 (v1.3), 9 (v1.4) og 10 (v1.5) lagt til. I tillegg kan brukere nå registrere sine egne symboler for egendefinerte operasjonseksporter og spesifisere dynamiske inngangsstørrelser under eksport.
  • tensorboard støtte er ikke lenger eksperimentell.
  • Lagt til nn.Transformermodul basert på artikkelen Oppmerksomhet er alt du trenger.
  • Tallrike forbedringer av C++ API.

Kilde: linux.org.ru

Legg til en kommentar