PyTorch, un popolare framework open source per l'apprendimento automatico, è stato aggiornato alla versione 1.2.0. La nuova versione include oltre 1900 correzioni che riguardano JIT, ONNX, modalità di apprendimento distribuito e miglioramenti delle prestazioni.
Alcuni cambiamenti:
- Nuova API TorchScript permette È facile convertire nn.Module (compresi i sottomoduli e i metodi chiamati in forward()) in ScriptModule.
- Insieme a Microsoft, è stato aggiunto il supporto completo per ONNX Opset versioni 7 (v1.2), 8 (v1.3), 9 (v1.4) e 10 (v1.5). Inoltre, gli utenti possono ora registrare i propri simboli per le esportazioni di operazioni personalizzate e specificare le dimensioni di input dinamiche durante l'esportazione.
- supporto tensorboard non è più sperimentale.
- Aggiunto modulo nn.Transformer in base all'articolo L'attenzione è tutto ciò che serve.
- Numerosi miglioramenti all'API C++.
Fonte: linux.org.ru