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.
>>> Подробности