PyTorch 1.2.0 llançat

PyTorch, un marc popular de codi obert per a l'aprenentatge automàtic, s'ha actualitzat a la versió 1.2.0. La nova versió inclou més de 1900 correccions que cobreixen JIT, ONNX, modes d'aprenentatge distribuïts i millores de rendiment.

Alguns canvis:

  • Nova API TorchScript permet És fàcil convertir nn.Module (inclosos submòduls i mètodes anomenats a forward()) a ScriptModule.
  • Juntament amb Microsoft, s'ha afegit suport complet per a les versions 7 (v1.2), 8 (v1.3), 9 (v1.4) i 10 (v1.5) d'ONNX Opset. A més, els usuaris ara poden registrar els seus propis símbols per a les exportacions d'operacions personalitzades i especificar mides d'entrada dinàmiques durant l'exportació.
  • suport tensorboard ja no és experimental.
  • S'ha afegit el mòdul nn.Transformer basat en l'article L'atenció és tot el que necessiteu.
  • Nombroses millores a l'API C++.

Font: linux.org.ru

Afegeix comentari