PyTorch 1.2.0 lançado

PyTorch, uma estrutura popular de código aberto para aprendizado de máquina, foi atualizada para a versão 1.2.0. A nova versão inclui mais de 1900 correções que cobrem JIT, ONNX, modos de aprendizagem distribuídos e melhorias de desempenho.

Algumas mudanças:

  • Nova API TorchScript permite É fácil converter nn.Module (incluindo submódulos e métodos chamados em forward()) em ScriptModule.
  • Juntamente com a Microsoft, foi adicionado suporte completo para ONNX Opset versões 7 (v1.2), 8 (v1.3), 9 (v1.4) e 10 (v1.5). Além disso, os usuários agora podem registrar seus próprios símbolos para exportações de operações personalizadas e especificar tamanhos de entrada dinâmicos durante a exportação.
  • suporte de tensorboard não é mais experimental.
  • Adicionado módulo nn.Transformer com base no artigo Atenção É Tudo Que Você Precisa.
  • Numerosas melhorias na API C++.

Fonte: linux.org.ru

Adicionar um comentário