Stable Baselines3

Overview

Stable Baselines3 (SB3) is a PyTorch-based library providing reliable implementations of reinforcement learning algorithms. This skill provides comprehensive guidance for training RL agents, creating custom environments, implementing callbacks, and optimizing training workflows using SB3's unified API.

When to Use This Skill

Use this skill when the user wants to:

• Train and evaluate a single-agent RL policy (PPO, A2C, SAC, TD3, DDPG, DQN, HER) on a Gymnasium environment.
• Wrap a research simulator as a custom gymnasium.Env and validate it with check_env.
• Add evaluation, checkpointing, early stopping, or custom logging through callbacks.
• Speed up training with vectorized environments, or save, load, and report trained agents reproducibly.

Does NOT Trigger