PyTorch, een populair open source-framework voor machine learning, is bijgewerkt naar versie 1.2.0. De nieuwe release bevat meer dan 1900 oplossingen voor JIT, ONNX, gedistribueerde leermodi en prestatieverbeteringen.
Enkele wijzigingen:
- Nieuwe TorchScript-API laat Het is gemakkelijk om nn.Module (inclusief submodules en methoden die in forward() worden aangeroepen) naar ScriptModule te converteren.
- Samen met Microsoft is volledige ondersteuning voor ONNX Opset versies 7 (v1.2), 8 (v1.3), 9 (v1.4) en 10 (v1.5) toegevoegd. Bovendien kunnen gebruikers nu hun eigen symbolen registreren voor exports van aangepaste bewerkingen en tijdens het exporteren dynamische invoergroottes opgeven.
- ondersteuning voor tensorboard is niet langer experimenteel.
- nn.Transformer-module toegevoegd op basis van het artikel Aandacht is alles wat je nodig hebt.
- Talrijke verbeteringen aan de C++ API.
Bron: linux.org.ru