Реализации алгоритмов обучения с подкреплением на PyTorch — надёжная и единообразная отправная точка для исследований и практических задач.
DLR-RM/rl-baselines3-zoo
Фреймворк для обучения агентов reinforcement learning на Stable Baselines3 с подобранными гиперпараметрами, предобученными агентами и скриптами для оптимизации гиперпараметров.
- Звёзды
- 2,900
- Форки
- 612
- Открытые issues
- 62
- Последний коммит
- 5 дней назад
Что это?
RL Baselines3 Zoo — фреймворк для обучения агентов reinforcement learning на базе Stable Baselines3. В него входят скрипты командной строки для обучения и оценки агентов, подбора гиперпараметров, построения графиков результатов и записи видео с агентами.
В проект также входит набор подобранных гиперпараметров для популярных сред и алгоритмов, а также агенты, обученные с этими настройками. Гиперпараметры для каждой среды заданы в YAML-файлах, поэтому запуск обучения сводится к одной команде с указанием алгоритма и ID среды. Обученных агентов можно запустить с помощью скрипта enjoy. Репозиторий охватывает среды Atari, classic control, Box2D, PyBullet, MuJoCo, robotics и MiniGrid.
Среди интеграций — Weights & Biases для отслеживания экспериментов и Hugging Face для хранения и публикации обученных моделей. Проект рассчитан на пользователей, которым нужен простой способ обучать RL-агентов, сравнивать алгоритмы и повторно использовать предобученные настройки.
Ключевые возможности
- 01Скрипты командной строки для обучения, оценки и повторного запуска агентов
- 02Подобранные гиперпараметры для каждой среды и алгоритма, заданные в YAML-файлах
- 03Инструменты для оптимизации гиперпараметров
- 04Предобученные агенты на Hugging Face
- 05Интеграции с Weights & Biases и Hugging Face
- 06Скрипты для построения графиков результатов обучения
Сценарии использования
- Обучение RL-агентов в средах Gym или PyBullet без написания шаблонного кода
- Сравнение разных RL-алгоритмов на одних и тех же задачах
- Поиск подходящих гиперпараметров для новой среды
- Повторное использование предобученных агентов для просмотра или оценки политик
Быстрый старт
- 1
Установите пакет rl_zoo3 из PyPI
pip install rl_zoo3 - 2
Обучите агента SAC в среде HalfCheetahBulletEnv-v0 с периодической оценкой (запускайте из клонированного репозитория)
python train.py --algo sac --env HalfCheetahBulletEnv-v0 --eval-freq 10000 --eval-episodes 10 --n-eval-envs 1
Что важно знать
- Для полной установки нужны системные пакеты, такие как swig, cmake и ffmpeg.
- В README предупреждают, что сборка pybullet в macOS может завершиться ошибкой без специального обходного решения с CFLAGS.
- Результаты бенчмарка получены в ходе отдельных запусков, поэтому они не являются количественным бенчмарком.
Темы
Частые вопросы
- Что такое rl-baselines3-zoo?
- Фреймворк для обучения агентов reinforcement learning на Stable Baselines3 с подобранными гиперпараметрами, предобученными агентами и скриптами для оптимизации гиперпараметров. RL Baselines3 Zoo — фреймворк для обучения агентов reinforcement learning на базе Stable Baselines3. В него входят скрипты командной строки для обучения и оценки агентов, подбора гиперпараметров, построения графиков результатов и записи видео с агентами.
- rl-baselines3-zoo можно использовать бесплатно? Какая у него лицензия?
- rl-baselines3-zoo распространяется с открытым исходным кодом под лицензией MIT. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- rl-baselines3-zoo всё ещё поддерживается?
- Активно поддерживается. Последний коммит в rl-baselines3-zoo — 16 окт. 2026 г..
- Для чего можно использовать rl-baselines3-zoo?
- Обучение RL-агентов в средах Gym или PyBullet без написания шаблонного кода Сравнение разных RL-алгоритмов на одних и тех же задачах Поиск подходящих гиперпараметров для новой среды Повторное использование предобученных агентов для просмотра или оценки политик
- Как установить rl-baselines3-zoo?
- Установите пакет rl_zoo3 из PyPI: pip install rl_zoo3 Обучите агента SAC в среде HalfCheetahBulletEnv-v0 с периодической оценкой (запускайте из клонированного репозитория): python train.py --algo sac --env HalfCheetahBulletEnv-v0 --eval-freq 10000 --eval-episodes 10 --n-eval-envs 1
- Где скачать rl-baselines3-zoo?
- Исходный код и инструкции по установке — на GitHub: https://github.com/DLR-RM/rl-baselines3-zoo
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
Открытый Python-фреймворк для применения глубокого обучения с подкреплением в автоматизированной торговле акциями и управлении портфелем.
Открытые VLA-модели для робототехники от Physical Intelligence: предобученные чекпойнты и примеры дообучения.
Стандартный Python API для одноагентных сред обучения с подкреплением и набор эталонных сред, форк OpenAI Gym.
Mamba — архитектура моделирования последовательностей на PyTorch, основанная на селективных моделях пространства состояний и предлагающая альтернативу Transformers с линейной сложностью для языковых и других задач.
Полный пайплайн предобучения с нуля и SFT небольшой китайской модели в стиле LLaMA2, который можно запустить на одной видеокарте с 24 ГБ памяти.