A PyTorch egy népszerű nyílt forráskódú keretrendszer a gépi tanuláshoz, amely frissült 1.2.0 verzióra. Az új kiadás több mint 1900 javítást tartalmaz, beleértve a JIT-et, az ONNX-t, a párhuzamos tanulási módokat és a teljesítményjavításokat.
Néhány változás:
- Új TorchScript API MMM-DD-YYYY könnyedén átkonvertálhatja az nn.Module-t (beleértve a beágyazott modulokat és a forward() metódusban hívott metódusokat) ScriptModule-ra.
- A Microsofttal együttműködve teljes támogatás került a 7-es (v1.2), 8-as (v1.3), 9-es (v1.4) és 10-es (v1.5) ONNX Opset verziókhoz. Ezen kívül a felhasználók most már regisztrálhatják saját szimbólumaikat egyedi műveletek exportálásához, és megadhatják a bemeneti adatok dinamikus méreteit exportálás közben.
- TensorBoard támogatás már nem kísérleti.
- Hozzáadva az nn.Transformer modul, amely a "Attention Is All You Need" cikk alapján készült. A figyelem minden, amire szükség van.
- Számos fejlesztés a C++ API-ban.
Forrás: linux.org.ru
