PyTorch, un framework open source populaire pour l'apprentissage automatique, a été mis à jour vers la version 1.2.0. La nouvelle version comprend plus de 1900 XNUMX correctifs couvrant JIT, ONNX, les modes d'apprentissage distribué et les améliorations de performances.
Quelques changements:
- Nouvelle API TorchScript il permet Il est facile de convertir nn.Module (y compris les sous-modules et les méthodes appelées dans forward()) en ScriptModule.
- En collaboration avec Microsoft, une prise en charge complète des versions 7 (v1.2), 8 (v1.3), 9 (v1.4) et 10 (v1.5) d'ONNX Opset a été ajoutée. De plus, les utilisateurs peuvent désormais enregistrer leurs propres symboles pour les exportations d'opérations personnalisées et spécifier des tailles d'entrée dynamiques lors de l'exportation.
- support de tableau tenseur n'est plus expérimental.
- Ajout du module nn.Transformer basé sur l'article L'attention est tout ce dont vous avez besoin.
- De nombreuses améliorations de l'API C++.
Source: linux.org.ru