PyTorch, popularna platforma open source do uczenia maszynowego, została zaktualizowana do wersji 1.2.0. Nowa wersja zawiera ponad 1900 poprawek obejmujących JIT, ONNX, tryby uczenia się rozproszonego i ulepszenia wydajności.
Trochę zmian:
- Nowe API TorchScript pozwala on Konwertowanie nn.Module (w tym podmodułów i metod wywoływanych w forward()) na ScriptModule jest łatwe.
- Razem z Microsoftem dodano pełną obsługę ONNX Opset w wersjach 7 (v1.2), 8 (v1.3), 9 (v1.4) i 10 (v1.5). Ponadto użytkownicy mogą teraz rejestrować własne symbole na potrzeby eksportu operacji niestandardowych i określać dynamiczne rozmiary danych wejściowych podczas eksportu.
- obsługa tensorboardu już nie jest eksperymentalny.
- Dodano moduł nn.Transformer na podstawie artykułu Uwaga jest wszystkim, czego potrzebujesz.
- Liczne ulepszenia API C++.
Źródło: linux.org.ru