Описание тега libtorch

Интерфейс PyTorch C++ обеспечивает высокоуровневый интерфейс моделирования на чистом C++ для нейронных сетей и общих сценариев использования в исследованиях и производстве машинного обучения, в значительной степени следуя API Python в дизайне и предоставляемых функциях. Интерфейс C++ включает следующее:

• Интерфейс для определения моделей машинного обучения через иерархическую модульную систему (например, torch.nn.Module);

• "Стандартная библиотека" уже существующих модулей для наиболее распространенных целей моделирования (например, свертки, RNN, пакетная нормализация и т. Д.);

• API оптимизации, включая реализации популярных оптимизаторов, таких как SGD, Adam, RMSprop и других;

• Средство представления наборов данных и конвейеров данных, включая функциональность для параллельной загрузки данных по многим ядрам ЦП;

• Формат сериализации для хранения и загрузки контрольных точек сеанса обучения (например, torch.utils.data.DataLoader);

• Автоматическое распараллеливание моделей на нескольких графических процессорах (например, torch.nn.parallel.DataParallel);

• Код поддержки для простой привязки моделей C++ к Python с помощью pybind11;

• Указывает на JIT-компилятор TorchScript;

• Полезные утилиты для облегчения взаимодействия с API Aten и Autograd.