37 lines
926 B
Bash
Executable File
37 lines
926 B
Bash
Executable File
#!/bin/bash
|
|
# 测试完整的MAGAIL训练流程
|
|
|
|
echo "======================================================================"
|
|
echo "🧪 测试MAGAIL完整训练流程"
|
|
echo "======================================================================"
|
|
|
|
# 测试参数(较小规模)
|
|
EPISODES=10
|
|
HORIZON=200
|
|
ROLLOUT=512
|
|
|
|
echo ""
|
|
echo "📋 测试配置:"
|
|
echo " Episodes: $EPISODES"
|
|
echo " Horizon: $HORIZON"
|
|
echo " Rollout Length: $ROLLOUT"
|
|
echo ""
|
|
|
|
# 运行训练(不渲染,加快速度)
|
|
python train_magail.py \
|
|
--episodes $EPISODES \
|
|
--horizon $HORIZON \
|
|
--rollout-length $ROLLOUT \
|
|
--batch-size 64 \
|
|
--lr-actor 3e-4 \
|
|
--lr-critic 3e-4 \
|
|
--lr-disc 3e-4 \
|
|
--epoch-disc 3 \
|
|
--epoch-ppo 5
|
|
|
|
echo ""
|
|
echo "======================================================================"
|
|
echo "✅ 测试完成"
|
|
echo "======================================================================"
|
|
|