Facebook опубликовала исходный код Pythia — это автоматически конфигурируемый модульный фреймворк глубокого обучения для обработки изображений и языка, сообщает VenutreBeat.
Pythia построен на базе открытого фреймворка PyTorch и позволяет разработчикам быстро строить, воссоздавать и сравнивать ИИ-модели. Фреймворк ориентирован на задачи по работе с визуальными и языковыми данными. Например может использоваться для ответов на вопросы, связанные с картинками, или генерирования аннотаций к ним.
Pythia включает несколько моделей (LoRRA, Pythia model и BAN) и имеет встроенную поддержку ряда датасетов (VQA, VizWiz, TextVQA и VisualDialog). Фреймоворк поддерживает одновременное обучение на нескольких датасетах, распределённое обучение, а также кастомные функции потерь, метрики, оптимизаторы и планирование.
Facebook обещает расширить первоначальный релиз новыми инструментами, задачами, датасетами и моделями. Исходный код Pythia доступен на GitHub.
Релоцировались? Теперь вы можете комментировать без верификации аккаунта.