PyTorch 1.2.0 lanzado

PyTorch, un marco popular de código aberto para a aprendizaxe automática, actualizouse á versión 1.2.0. A nova versión inclúe máis de 1900 correccións que abarcan JIT, ONNX, modos de aprendizaxe distribuídos e melloras de rendemento.

Algúns cambios:

  • Nova API de TorchScript permite É doado converter nn.Module (incluíndo submódulos e métodos chamados en forward()) a ScriptModule.
  • Xunto con Microsoft, engadiuse soporte completo para as versións 7 (v1.2), 8 (v1.3), 9 (v1.4) e 10 (v1.5) de ONNX Opset. Ademais, agora os usuarios poden rexistrar os seus propios símbolos para exportacións de operacións personalizadas e especificar tamaños de entrada dinámicos durante a exportación.
  • soporte de placa tensor xa non está experimental.
  • Engadido módulo nn.Transformer baseado no artigo A atención é todo o que necesitas.
  • Numerosas melloras na API de C++.

Fonte: linux.org.ru

Engadir un comentario