Megjelent a PyTorch 1.2.0

A PyTorch, a gépi tanulás népszerű nyílt forráskódú keretrendszere az 1.2.0-s verzióra frissült. Az új kiadás több mint 1900 javítást tartalmaz, amelyek lefedik a JIT-et, az ONNX-et, az elosztott tanulási módokat és a teljesítménybeli fejlesztéseket.

Néhány változás:

  • Új TorchScript API lehetővé teszi Könnyen konvertálható az nn.Module (beleértve az almodulokat és a forward()-ban meghívott metódusokat) ScriptModule-ra.
  • A Microsofttal együtt teljes mértékben támogatja az ONNX Opset 7-es (v1.2), 8-as (v1.3), 9-es (v1.4) és 10-es (v1.5) verzióját. Ezenkívül a felhasználók regisztrálhatják saját szimbólumaikat az egyéni műveletek exportálásához, és megadhatják a dinamikus bemeneti méreteket az exportálás során.
  • tenzortábla támaszték többé nem kísérleti.
  • nn.Transformer modul hozzáadva a cikk alapján Csak a figyelem kell.
  • A C++ API számos fejlesztése.

Forrás: linux.org.ru

Hozzászólás