PyTorch, ein beliebtes Open-Source-Framework für maschinelles Lernen, wurde auf Version 1.2.0 aktualisiert. Die neue Version enthält mehr als 1900 Korrekturen, die JIT, ONNX, verteilte Lernmodi und Leistungsverbesserungen abdecken.
Einige Veränderungen:
- Neue TorchScript-API ermöglicht Es ist einfach, nn.Module (einschließlich der in forward() aufgerufenen Submodule und Methoden) in ScriptModule zu konvertieren.
- Zusammen mit Microsoft wurde die vollständige Unterstützung für die ONNX Opset-Versionen 7 (v1.2), 8 (v1.3), 9 (v1.4) und 10 (v1.5) hinzugefügt. Darüber hinaus können Benutzer jetzt ihre eigenen Symbole für benutzerdefinierte Operationsexporte registrieren und während des Exports dynamische Eingabegrößen angeben.
- Tensorboard-Unterstützung ist nicht mehr Experimental.
- nn.Transformer-Modul basierend auf dem Artikel hinzugefügt Aufmerksamkeit ist alles was Sie brauchen.
- Zahlreiche Verbesserungen an der C++-API.
Source: linux.org.ru