PyTorch — популярный фреймворк для машинного обучения — обновился до версии 1.5.0. Этот релиз включает в себя несколько крупных дополнений и улучшений API, среди которых можно отметить:
-
C++ API, ранее считавшийся экспериментальным, окончательно стабилизирован. Теперь пользователи могут легко переводить свои модели с Python API на C++ API.
-
Стабилизирован пакет
torch.distributed.rpc
, предоставляющий широкие возможности в распределенном обучении, включая автоматическое вычисление градиентов и обновление параметров модели. -
Обновлен torch_xla — пакет, который использует компилятор XLA для ускорения обучения моделей на облачных TPU.
-
Также обновлены пакеты torchaudio, torchvision и torchtext, предоставляющие инструменты для разработки моделей, обрабатывающих звуковые, графические и текстовые данные.
-
Python 2 больше не поддерживается. Вся дальнейшая разработка будет вестись только для Python 3.
>>> Подробности