What problem does it solve?
Stable Baselines3 work can be time-consuming when you need to choose an algorithm, validate a custom Gymnasium environment, monitor training, save checkpoints, and evaluate results consistently. This Skill turns those repeated reinforcement learning setup tasks into a clear, reliable workflow for experimentation and prototyping.
Core Features & Use Cases
- Algorithm Guidance: Helps select and configure PPO, A2C, SAC, TD3, DQN, and HER for the right action space and task type.
- Environment and Training Support: Covers custom Gymnasium environments, vectorized environments, normalization, callbacks, and environment validation.
- Evaluation and Persistence: Supports model saving and loading, evaluation, and video recording for training runs and benchmark comparisons.
- Use Case: A researcher building a new navigation task can validate the environment, train with vectorized workers, checkpoint the best model, and compare results across algorithms.
Quick Start
Ask the Skill to help you design, validate, train, and evaluate a Stable Baselines3 reinforcement learning agent for your Gymnasium environment.