Facebook открыла исходный код фреймворка глубокого обучения Pythia

Facebook опубликовала исходный код Pythia — это автоматически конфигурируемый модульный фреймворк глубокого обучения для обработки изображений и языка, сообщает VenutreBeat.

Pythia построен на базе открытого фреймворка PyTorch и позволяет разработчикам быстро строить, воссоздавать и сравнивать ИИ-модели. Фреймворк ориентирован на задачи по работе с визуальными и языковыми данными. Например может использоваться для ответов на вопросы, связанные с картинками, или генерирования аннотаций к ним.

Pythia включает несколько моделей (LoRRA, Pythia model и BAN) и имеет встроенную поддержку ряда датасетов (VQA, VizWiz, TextVQA и VisualDialog). Фреймоворк поддерживает одновременное обучение на нескольких датасетах, распределённое обучение, а также кастомные функции потерь, метрики, оптимизаторы и планирование.

Facebook обещает расширить первоначальный релиз новыми инструментами, задачами, датасетами и моделями. Исходный код Pythia доступен на GitHub.