PyTorch – en populär öppen ram för maskininlärning har uppdaterats till version 1.2.0. Den nya versionen innehåller över 1900 buggfixar som täcker JIT, ONNX, distribuerade träningslägen och prestandaförbättringar.
Några ändringar:
- Nytt TorchScript API möjliggör lätt att konvertera nn.Module (inklusive undermoduler och metoder som kallas i forward()) till ScriptModule.
- Tillsammans med Microsoft har full stöd för ONNX Opset-versioner 7 (v1.2), 8 (v1.3), 9 (v1.4) och 10 (v1.5) lagts till. Dessutom kan användare registrera sina egna symboler för export av anpassade operationer och ange dynamiska storlekar på ingående data vid export.
- Stöd för tensorboard är inte längre experimentell.
- Modulen nn.Transformer har lagts till, baserat på artikeln Attention Is All You Need.
- Många förbättringar i C++ API.
Källa: linux.org.ru
