What problem does it solve?
Stable Baselines3 provides production-ready reinforcement learning algorithms with a unified PyTorch API, enabling reliable training, quick prototyping, and accessible experimentation for developers and researchers.
Core Features & Use Cases
- Training RL agents with PPO, SAC, DQN, TD3, DDPG, A2C via SB3's unified API for single-agent tasks and reproducible experiments.
- Custom Environments: guidelines and templates for Gymnasium environments, with validation and best practices for environment design.
- Vectorized Environments: support through make_vec_env, DummyVecEnv, and SubprocVecEnv to accelerate training.
- Callbacks for Monitoring and Control: EvalCallback, CheckpointCallback, StopTrainingOnRewardThreshold, and more for robust training pipelines.
- Model Persistence and Evaluation: save/load models, normalize statistics, and evaluate performance with evaluate_policy.
- Workflow and References: template scripts and reference guides to streamline RL project work.
Quick Start
Install stable-baselines3 and run the training template (scripts/train_rl_agent.py) to kick off training.