What problem does it solve?
It helps you implement and debug reinforcement learning training workflows by providing reliable, end-to-end guidance for using Stable Baselines3 with Gymnasium-style environments.
Core Features & Use Cases
- Algorithm implementation guidance: Choose and configure core SB3 algorithms (PPO, SAC, TD3, DQN, A2C) for discrete/continuous and on-policy/off-policy scenarios, including practical hyperparameter tips and when to use HER or RecurrentPPO.
- Custom environment design: Build Gymnasium-compatible environments (spaces, reset/step signatures, termination vs truncation) and validate them with SB3’s
check_env() to prevent subtle interface bugs.
- Vectorized training & callbacks: Accelerate data collection with DummyVecEnv/SubprocVecEnv, handle VecEnv API differences, and structure training monitoring via Eval/Checkpoint callbacks, model saving/loading, and evaluation patterns (including recording video).
Quick Start
Use the stable-baselines3 skill to set up and validate a custom Gymnasium environment, select the appropriate SB3 RL algorithm, and write a minimal training loop with callbacks for evaluation and checkpointing.