PyTorch, populárny open source framework pre strojové učenie, bol aktualizovaný na verziu 1.2.0. Nové vydanie obsahuje viac ako 1900 XNUMX opráv, ktoré pokrývajú JIT, ONNX, distribuované režimy učenia a vylepšenia výkonu.
Niektoré zmeny:
- Nové TorchScript API umožňuje Je ľahké previesť nn.Module (vrátane podmodulov a metód volaných v forward()) na ScriptModule.
- Spolu s Microsoftom bola pridaná plná podpora pre ONNX Opset verzie 7 (v1.2), 8 (v1.3), 9 (v1.4) a 10 (v1.5). Okrem toho môžu používatelia teraz zaregistrovať svoje vlastné symboly pre exporty vlastných operácií a špecifikovať dynamické vstupné veľkosti počas exportu.
- podpora tenzorovej dosky už nie je experimentálne.
- Pridaný modul nn.Transformer na základe článku Pozornosť je všetko, čo potrebujete.
- Početné vylepšenia C++ API.
Zdroj: linux.org.ru