PyTorch — un framework open source populaire pour l'apprentissage automatique a été mis à jour vers la version 1.2.0. Cette nouvelle version comprend plus de 1900 correctifs couvrant JIT, ONNX, les modes d'apprentissage distribués et des améliorations de performances.
Quelques changements :
- Nouvelle API TorchScript d'installer permet de convertir facilement nn.Module (y compris les sous-modules et les méthodes appelées dans forward()) en ScriptModule.
- En collaboration avec Microsoft, un support complet des ONNX Opset pour les versions 7(v1.2), 8(v1.3), 9(v1.4) et 10 (v1.5) a été ajouté. De plus, les utilisateurs peuvent désormais enregistrer leurs propres symboles pour l'exportation d'opérations personnalisées et indiquer des tailles dynamiques pour les entrées lors de l'exportation.
- Support de tensorboard n'est plus expérimental.
- Ajouté le module nn.Transformer, basé sur l'article Attention Is All You Need.
- De nombreuses améliorations dans l'API C++.
Source : linux.org.ru
