DLR-RM/rl-baselines3-zoo

O que é?

Recursos principais

• 01Scripts de linha de comando para treinar, avaliar e reproduzir agentes
• 02Hiperparâmetros ajustados por ambiente e algoritmo, definidos em arquivos YAML
• 03Ferramentas para otimização de hiperparâmetros
• 04Agentes pré-treinados disponíveis no Hugging Face
• 05Integrações com Weights & Biases e Hugging Face
• 06Scripts para gerar gráficos dos resultados de treinamento

Casos de uso

• Treinar agentes de RL em ambientes Gym ou PyBullet sem escrever código boilerplate
• Comparar diferentes algoritmos de RL nas mesmas tarefas
• Buscar bons hiperparâmetros para um novo ambiente
• Reutilizar agentes pré-treinados para assistir às políticas ou avaliá-las

Início rápido

1. 1

   Instale o pacote rl_zoo3 do PyPI

   pip install rl_zoo3
2. 2

   Treine um agente SAC no HalfCheetahBulletEnv-v0 com avaliações periódicas (execute a partir do repositório clonado)

   python train.py --algo sac --env HalfCheetahBulletEnv-v0 --eval-freq 10000 --eval-episodes 10 --n-eval-envs 1

Antes de usar

• !A instalação completa requer pacotes do sistema, como swig, cmake e ffmpeg.
• !O README alerta que o pybullet pode falhar ao compilar no macOS sem uma solução alternativa especial para CFLAGS.
• !Os resultados do benchmark vêm de execuções únicas e, portanto, não constituem um benchmark quantitativo.

Tópicos

Perguntas frequentes

O que é rl-baselines3-zoo?

Framework de treinamento para agentes de aprendizado por reforço do Stable Baselines3, com hiperparâmetros ajustados, agentes pré-treinados e scripts de otimização de hiperparâmetros. O RL Baselines3 Zoo é um framework de treinamento para aprendizado por reforço baseado no Stable Baselines3. Ele oferece scripts de linha de comando para treinar e avaliar agentes, ajustar hiperparâmetros, gerar gráficos de resultados e gravar vídeos dos agentes.

rl-baselines3-zoo é gratuito? Qual é a licença?

rl-baselines3-zoo é open source sob a licença MIT. Confira os termos da licença antes de usá-lo em produtos comerciais.

rl-baselines3-zoo ainda é mantido?

Manutenção ativa. O último commit de rl-baselines3-zoo foi em 16 de out. de 2026.

Para que posso usar rl-baselines3-zoo?

Treinar agentes de RL em ambientes Gym ou PyBullet sem escrever código boilerplate Comparar diferentes algoritmos de RL nas mesmas tarefas Buscar bons hiperparâmetros para um novo ambiente Reutilizar agentes pré-treinados para assistir às políticas ou avaliá-las

Como instalar rl-baselines3-zoo?

Instale o pacote rl_zoo3 do PyPI: pip install rl_zoo3 Treine um agente SAC no HalfCheetahBulletEnv-v0 com avaliações periódicas (execute a partir do repositório clonado): python train.py --algo sac --env HalfCheetahBulletEnv-v0 --eval-freq 10000 --eval-episodes 10 --n-eval-envs 1

Onde consigo rl-baselines3-zoo?

O código-fonte e as instruções de instalação estão no GitHub: https://github.com/DLR-RM/rl-baselines3-zoo