从安装到训练
按下列步骤可在本仓库复现完整实验。建议先跑 analytical,再跑 genesis。
-
创建环境(Python 3.11)
cd RL_Demo uv venv --python 3.11 .venv source .venv/bin/activate uv pip install torch --index-url https://download.pytorch.org/whl/cpu uv pip install -r requirements.txt -
快速训练(分析动力学)
python scripts/train.py --backend analytical --iterations 200 -
Genesis 训练
python scripts/train.py --backend genesis --num-envs 8 --iterations 100CPU 上 Genesis 较慢,可先减小 num_envs / iterations。
-
评估策略
python scripts/evaluate.py --backend analytical python scripts/evaluate.py --backend genesis -
打开本教学站
python scripts/serve_website.py
训练循环伪代码
for iteration in range(N):
for t in range(steps_per_env):
a, logp, V = policy(obs)
obs, r, done, info = env.step(a)
buffer.store(...)
advantages, returns = GAE(buffer)
for epoch in range(K):
for minibatch in buffer:
update PPO clip loss
关键超参(见 configs/train_config.yaml)
| 参数 | 默认 | 含义 |
|---|---|---|
| gamma | 0.99 | 折扣 |
| gae_lambda | 0.95 | GAE 平滑 |
| clip_eps | 0.2 | PPO 裁剪 |
| learning_rate | 3e-4 | Adam 学习率 |
| steps_per_env | 128 | 每次更新前采样长度 |
| num_envs | 32 | 并行环境 |