PyTorch, ett populärt ramverk med öppen källkod för maskininlärning, har uppdaterats till version 1.2.0. Den nya utgåvan innehåller mer än 1900 XNUMX korrigeringar som täcker JIT, ONNX, distribuerade inlärningslägen och prestandaförbättringar.
Några förändringar:
- Nytt TorchScript API det gör Det är lätt att konvertera nn.Module (inklusive undermoduler och metoder som anropas i forward()) till ScriptModule.
- Tillsammans med Microsoft har fullt stöd för ONNX Opset version 7 (v1.2), 8 (v1.3), 9 (v1.4) och 10 (v1.5) lagts till. Dessutom kan användare nu registrera sina egna symboler för anpassade operationsexporter och ange dynamiska indatastorlekar under export.
- stöd för tensorboard är inte längre experimentell.
- Lade till nn.Transformer-modul baserad på artikeln Uppmärksamhet är allt du behöver.
- Många förbättringar av C++ API.
Källa: linux.org.ru