Repolode
DLR-RM

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. 1

    Установите пакет rl_zoo3 из PyPI

    pip install rl_zoo3
  2. 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 обновляется ежедневно.

Похожие проекты

Mamba — архитектура моделирования последовательностей на PyTorch, основанная на селективных моделях пространства состояний и предлагающая альтернативу Transformers с линейной сложностью для языковых и других задач.

18,9 тыс.1,8 тыс.PythonApache-2.0Обновлён 3 месяца назад