PyTorch — populární open-source framework pro strojové učení byl aktualizován na verzi 1.2.0. Nová verze obsahuje více než 1900 oprav, které pokrývají JIT, ONNX, distribuované režimy učení a zlepšení výkonu.
Některé změny:
- Nové API TorchScript umožňuje snadno převést nn.Module (včetně podmodulů a metod volaných v forward()) na ScriptModule.
- Ve spolupráci se společností Microsoft byla přidána plná podpora pro ONNX Opset verzí 7(v1.2), 8(v1.3), 9(v1.4) a 10 (v1.5). Navíc nyní mohou uživatelé registrovat své vlastní symboly pro export vlastních operací a uvádět dynamické rozměry vstupních dat při exportu.
- Podpora tensorboard už není experimentální.
- Byl přidán modul nn.Transformer založený na článku Attention Is All You Need.
- Několik vylepšení v C++ API.
Zdroj: linux.org.ru
