修改了算法代码,并建立了一个简单的训练脚本.修改bert处理二维输入,移除PPO的permute参数
This commit is contained in:
23
快速测试.sh
Executable file
23
快速测试.sh
Executable 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 "=========================================="
|
||||
|
||||
Reference in New Issue
Block a user