PyTorch — um popular framework open-source para aprendizado de máquina foi atualizado para a versão 1.2.0. A nova versão inclui mais de 1900 correções, que abrangem JIT, ONNX, modos de treinamento distribuído e melhorias de desempenho.
Algumas mudanças:
- Nova API TorchScript permite facilmente converter nn.Module (incluindo submódulos e métodos chamados em forward()) em ScriptModule.
- Junto com a Microsoft, foi adicionada suporte completo para ONNX Opset das versões 7(v1.2), 8(v1.3), 9(v1.4) e 10 (v1.5). Além disso, agora os usuários podem registrar seus próprios símbolos para exportar operações personalizadas e especificar tamanhos dinâmicos de entrada durante a exportação.
- Suporte ao tensorboard não é mais experimental.
- Adicionado o módulo nn.Transformer, baseado no artigo Attention Is All You Need.
- Numerosas melhorias na API C++.
Fonte: linux.org.ru
