🚀 PPO 智能體玩轉 SpaceInvadersNoFrameskip-v4
本項目是一個經過訓練的 PPO 智能體,它使用 stable-baselines3 庫 和 RL Zoo 來玩 SpaceInvadersNoFrameskip-v4 遊戲。RL Zoo 是 Stable Baselines3 強化學習智能體的訓練框架,包含超參數優化和預訓練智能體。
🚀 快速開始
模型下載與運行
# 下載模型並保存到 logs/ 文件夾
python -m rl_zoo3.load_from_hub --algo ppo --env SpaceInvadersNoFrameskip-v4 -orga sb3 -f logs/
python enjoy.py --algo ppo --env SpaceInvadersNoFrameskip-v4 -f logs/
模型訓練與上傳
python train.py --algo ppo --env SpaceInvadersNoFrameskip-v4 -f logs/
# 上傳模型並生成視頻(如果可能)
python -m rl_zoo3.push_to_hub --algo ppo --env SpaceInvadersNoFrameskip-v4 -f logs/ -orga sb3
✨ 主要特性
本項目使用 PPO 算法訓練智能體,使其能夠在 SpaceInvadersNoFrameskip-v4 遊戲中表現出色。同時,藉助 RL Zoo 框架,可以方便地進行超參數優化和模型訓練。
📦 安裝指南
本項目依賴以下庫:
- RL Zoo: https://github.com/DLR-RM/rl-baselines3-zoo
- SB3: https://github.com/DLR-RM/stable-baselines3
- SB3 Contrib: https://github.com/Stable-Baselines-Team/stable-baselines3-contrib
請確保在運行代碼前安裝好這些依賴。
💻 使用示例
基礎用法
# 下載模型並保存到 logs/ 文件夾
python -m rl_zoo3.load_from_hub --algo ppo --env SpaceInvadersNoFrameskip-v4 -orga sb3 -f logs/
python enjoy.py --algo ppo --env SpaceInvadersNoFrameskip-v4 -f logs/
高級用法
python train.py --algo ppo --env SpaceInvadersNoFrameskip-v4 -f logs/
# 上傳模型並生成視頻(如果可能)
python -m rl_zoo3.push_to_hub --algo ppo --env SpaceInvadersNoFrameskip-v4 -f logs/ -orga sb3
📚 詳細文檔
超參數
OrderedDict([('batch_size', 256),
('clip_range', 'lin_0.1'),
('ent_coef', 0.01),
('env_wrapper',
['stable_baselines3.common.atari_wrappers.AtariWrapper']),
('frame_stack', 4),
('learning_rate', 'lin_2.5e-4'),
('n_envs', 8),
('n_epochs', 4),
('n_steps', 128),
('n_timesteps', 10000000.0),
('policy', 'CnnPolicy'),
('vf_coef', 0.5),
('normalize', False)])
模型指標
屬性 |
詳情 |
模型類型 |
PPO |
訓練數據 |
SpaceInvadersNoFrameskip-v4 |
平均獎勵 |
886.50 +/- 417.30 |