修改了算法代码,并建立了一个简单的训练脚本.修改bert处理二维输入,移除PPO的permute参数

This commit is contained in:
2025-10-22 16:56:12 +08:00
parent b626702cbb
commit 3f7e183c4b
101 changed files with 3837 additions and 39 deletions

23
快速测试.sh Executable file
View File

@@ -0,0 +1,23 @@
#!/bin/bash
# 快速测试脚本
echo "=========================================="
echo "测试 MAGAIL 训练脚本"
echo "=========================================="
# 激活conda环境并运行
cd /home/huangfukk/MAGAIL4AutoDrive
echo ""
echo "1. 测试模块导入..."
python -c "from Algorithm.magail import MAGAIL; print('✅ 导入成功!')" || exit 1
echo ""
echo "2. 运行训练1个episode不渲染..."
python train_magail.py --episodes 1 --horizon 50 2>&1 | tail -30
echo ""
echo "=========================================="
echo "测试完成!"
echo "=========================================="