PyTorch, un popular marco de código abierto para aprendizaje automático, se actualizó a la versión 1.2.0. La nueva versión incluye más de 1900 correcciones que cubren JIT, ONNX, modos de aprendizaje distribuido y mejoras de rendimiento.
Algunos cambios:
- Nueva API de TorchScript permite Es fácil convertir nn.Module (incluidos los submódulos y métodos llamados en forward()) a ScriptModule.
- Junto con Microsoft, se ha agregado soporte completo para ONNX Opset versiones 7 (v1.2), 8 (v1.3), 9 (v1.4) y 10 (v1.5). Además, los usuarios ahora pueden registrar sus propios símbolos para exportaciones de operaciones personalizadas y especificar tamaños de entrada dinámicos durante la exportación.
- soporte de tensorboard ya no es experimental.
- Se agregó el módulo nn.Transformer basado en el artículo. La atención es todo lo que necesitas.
- Numerosas mejoras en la API de C++.
Fuente: linux.org.ru