PyTorch — popularny otwarty framework do uczenia maszynowego zaktualizował się do wersji 1.2.0. Nowa wersja zawiera ponad 1900 poprawek, które obejmują JIT, ONNX, tryby uczenia rozproszonego oraz poprawę wydajności.
Niektóre zmiany:
- Nowe API TorchScript przekazywanie łatwo skonwertować nn.Module (w tym podmoduły i metody wywoływane w forward()) na ScriptModule.
- Wspólnie z Microsoft dodano pełne wsparcie dla ONNX Opset wersji 7(v1.2), 8(v1.3), 9(v1.4) i 10 (v1.5). Co więcej, użytkownicy mogą teraz rejestrować własne symbole do eksportu niestandardowych operacji i określać dynamiczne rozmiary danych wejściowych podczas eksportu.
- Wsparcie dla tensorboard nie jest już eksperymentalne.
- Dodano moduł nn.Transformer oparty na artykule Attention Is All You Need.
- Liczne ulepszenia w API C++.
Źródło: linux.org.ru
