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 XNUMX oprav, které pokrývají JIT, ONNX, režimy distribuovaného učení a vylepšení výkonu.
Některé změny:
- Nové TorchScript API umožňuje Je snadné převést nn.Module (včetně submodulů a metod volaných v forward()) na ScriptModule.
- Společně s Microsoftem byla přidána plná podpora pro ONNX Opset verze 7 (v1.2), 8 (v1.3), 9 (v1.4) a 10 (v1.5). Uživatelé nyní mohou navíc registrovat své vlastní symboly pro exporty vlastních operací a specifikovat dynamické vstupní velikosti během exportu.
- podpora tenzorové desky již není více experimentální.
- Přidán modul nn.Transformer na základě článku Pozornost je vše, co potřebujete.
- Četná vylepšení C++ API.
Zdroj: linux.org.ru