What problem does it solve?
This Skill reduces the boilerplate and common pitfalls of training, evaluating, and persisting reinforcement learning agents by providing templates, best practices, and comprehensive references for Stable Baselines3 so practitioners can iterate faster and avoid integration errors.
Core Features & Use Cases
- Training templates: A production-ready training script with vectorized environments, TensorBoard integration, and checkpointing for common algorithms.
- Evaluation & recording: Evaluation and video-recording utilities to measure performance, load VecNormalize statistics, and compare models.
- Custom environments & validation: A full custom Gymnasium environment template and guidance on implementing, validating, and registering environments compatible with SB3.
- Vectorization & callbacks: Guidance and examples for DummyVecEnv/SubprocVecEnv, VecNormalize, callback chaining, early stopping, and custom monitoring.
- Model persistence & inspection: Best practices for saving/loading models, normalization stats, and accessing parameters/state dicts.
Quick Start
Train a PPO agent on CartPole using the provided training template to obtain a working model, evaluation logs, and saved checkpoints.