Вышел PyTorch 1.2.0

PyTorch — популярный открытый фреймворк для машинного обучения обновился до версии 1.2.0. Новый релиз включает в себя более 1900 исправлений, которые охватывают JIT, ONNX, распределенные режимы обучения и улучшения производительности.

Некоторые изменения:

  • Новый TorchScript API позволяет легко сконвертировать nn.Module (включая подмодули и методы, вызываемые в forward()) в ScriptModule.
  • Совместно с Microsoft добавлена полная поддержка ONNX Opset версий 7(v1.2), 8(v1.3), 9(v1.4) и 10 (v1.5). Кроме того, теперь пользователи могут зарегистрировать свои собственные символы для экспорта кастомных операций и указывать динамические размеры входных данных во время экспорта.
  • Поддержка tensorboard больше не является экспериментальной.
  • Добавлен модуль nn.Transformer, основанный на статье Attention Is All You Need.
  • Многочисленные улучшения в C++ API.

Источник: linux.org.ru

author avatar
ProHoster Консультант, Технический специалист
A technical specialist at ProHoster with over six years of experience in server administration, VPN solutions, and network security. I manage infrastructure setup and support, monitor service stability, and implement solutions to protect client data. I also contribute to performance optimization and compliance with modern security and privacy requirements.

Добавить комментарий