Фреймворк с определениями современных моделей машинного обучения для текста, изображений, аудио и мультимодальных задач — для обучения и инференса.
google-research/text-to-text-transfer-transformer
Кодовая база Google Research для воспроизведения экспериментов «Text-to-Text Transfer Transformer»: предобработка данных, обучение, дообучение и оценка на TPU.
- Звёзды
- 6,565
- Форки
- 800
- Открытые issues
- 110
- Последний коммит
- 26 дней назад
Что это?
T5 — это кодовая база к статье «Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer». В ней каждая задача NLP представлена в формате «текст на входе — текст на выходе», а библиотека в основном предназначена для загрузки, предобработки, объединения и оценки датасетов. Пользователи создают Tasks на основе TensorFlow Datasets, текстовых файлов или собственных источников данных, задают препроцессоры, токенизацию SentencePiece и метрики, а затем объединяют задачи в Mixtures для многозадачного обучения.
В репозитории есть исполняемые файлы командной строки, например t5_mesh_transformer, для предобучения, дообучения, оценки и декодирования; их работа настраивается с помощью файлов конфигурации gin. В первую очередь проект рассчитан на Google Cloud TPU с Mesh TensorFlow; также есть инструкции по запуску на GPU. Кроме того, доступен экспериментальный адаптер Hugging Face Transformers для дообучения с PyTorch на одном GPU, а также опубликованные чекпойнты от T5-Small до T5-11B.
Проект больше не развивается активно; в README для новых задач рекомендуется использовать реализацию T5X на JAX и Flax.
Ключевые возможности
- 01Представляет задачи NLP как данные в формате «текст на входе — текст на выходе» с помощью Task, TfdsTask и TextLineTask.
- 02Объединяет несколько задач в Mixtures для многозадачного обучения с настраиваемыми коэффициентами.
- 03Предоставляет инструменты командной строки для предобучения, дообучения, оценки и декодирования.
- 04Работает на Google Cloud TPU с Mesh TensorFlow; инструкции по запуску на GPU также включены.
- 05Публикует предобученные чекпойнты от T5-Small (60M) до T5-11B (11B параметров).
Сценарии использования
- Воспроизведение экспериментов по трансферному обучению из статьи о T5.
- Дообучение опубликованных чекпойнтов T5 на собственном датасете в формате «текст на входе — текст на выходе».
- Создание собственных конвейеров многозадачного обучения NLP с объединением датасетов.
- Изучение целей предобучения и предобработки датасетов для seq2seq-моделей.
Быстрый старт
- 1
Установить пакет T5
pip install t5[gcp] - 2
Попробовать T5 в бесплатном руководстве Colab (TPU, локальная настройка не требуется)
- 3
Просмотреть доступные задачи и наборы смесей
python -c "import t5; print(t5.data.MixtureRegistry.names())"
Что важно знать
- В README указано, что T5 на TensorFlow с MeshTF больше не разрабатывается активно; вместо него рекомендуется T5X.
- Обучение и дообучение рассчитаны на использование TPU Google Cloud и хранилища GCP.
- Подготовка набора данных C4 требует загрузки около 7 ТБ и примерно 335 дней вычислений на CPU.
В подборках
Темы
Частые вопросы
- Что такое text-to-text-transfer-transformer?
- Кодовая база Google Research для воспроизведения экспериментов «Text-to-Text Transfer Transformer»: предобработка данных, обучение, дообучение и оценка на TPU. T5 — это кодовая база к статье «Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer». В ней каждая задача NLP представлена в формате «текст на входе — текст на выходе», а библиотека в основном предназначена для загрузки, предобработки, объединения и оценки датасетов. Пользователи создают Tasks на основе TensorFlow Datasets, текстовых файлов или собственных источников данных, задают препроцессоры, токенизацию SentencePiece и метрики, а затем объединяют задачи в Mixtures для многозадачного обучения.
- text-to-text-transfer-transformer можно использовать бесплатно? Какая у него лицензия?
- text-to-text-transfer-transformer распространяется с открытым исходным кодом под лицензией Apache-2.0. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- text-to-text-transfer-transformer всё ещё поддерживается?
- Активно поддерживается. Последний коммит в text-to-text-transfer-transformer — 12 окт. 2026 г..
- Для чего можно использовать text-to-text-transfer-transformer?
- Воспроизведение экспериментов по трансферному обучению из статьи о T5. Дообучение опубликованных чекпойнтов T5 на собственном датасете в формате «текст на входе — текст на выходе». Создание собственных конвейеров многозадачного обучения NLP с объединением датасетов. Изучение целей предобучения и предобработки датасетов для seq2seq-моделей.
- Как установить text-to-text-transfer-transformer?
- Установить пакет T5: pip install t5[gcp] Попробовать T5 в бесплатном руководстве Colab (TPU, локальная настройка не требуется) Просмотреть доступные задачи и наборы смесей: python -c "import t5; print(t5.data.MixtureRegistry.names())"
- Где скачать text-to-text-transfer-transformer?
- Исходный код и инструкции по установке — на GitHub: https://github.com/google-research/text-to-text-transfer-transformer
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
Компоненты и архитектуры spaCy для интеграции в пайплайны spaCy моделей-трансформеров Hugging Face, таких как BERT, RoBERTa и XLNet.
Библиотека Python и CLI от Deezer, которая разделяет песни на вокал, ударные, бас и другие дорожки с помощью предварительно обученных моделей.
Инструментарий на Python для разбиения текста на предложения и другие смысловые фрагменты с предобученными моделями SaT и WtP для 85 языков.
Библиотека предобученных моделей для Keras 3: модели для текста, изображений и аудио, работающие с JAX, TensorFlow или PyTorch.
Лёгкая библиотека глубокого обучения с проверкой типов для функциональной композиции моделей и обёртывания слоёв PyTorch, TensorFlow и MXNet.