stable-baselines3 - 生产级强化学习算法库
基于 PyTorch 的强化学习算法库，支持训练智能体和创建自定义环境
标签
更新于: 2026-06-30能力
典型输入
典型输出
该技能可以做什么
- 训练强化学习智能体
- 创建自定义环境
- 实现回调函数
- 保存模型
- 加载模型
- 评估智能体
- 录制视频
- 向量化环境
- 监控训练指标
- 优化训练流程
输入
- Gymnasium 环境
- 算法类型
- 训练超参数
- 回调函数定义
- 策略配置
- 学习率调度
输出
- 训练好的模型文件
- 评估指标
- 视频录制
- TensorBoard 日志
- 模型检查点
要求
- Python 环境
- PyTorch
- Gymnasium
- stable-baselines3 包