blob: 43763df3d41694ad1bdf8511149c9bba52ce9b0e (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
#!/bin/bash
# GEE快速测试脚本
# 使用合成数据进行快速验证，无需真实模型
echo "开始GEE快速测试..."
# 设置环境变量
export CUDA_VISIBLE_DEVICES=0
echo "运行组件测试..."
python test_gee_components.py
if [ $? -eq 0 ]; then
echo "✓ 组件测试通过"
else
echo "✗ 组件测试失败"
exit 1
fi
echo ""
echo "运行快速训练测试（使用合成数据）..."
accelerate launch train_gee.py \
--model_name Qwen2.5-Math-7B \
--model_path /volume/pt-train/models/Qwen2.5-Math-7B \
--use_test_data \
--effective_batch 8 \
--micro_batch_size 2 \
--max_steps 5 \
--lambda_weight 3.0 \
--log_steps 1 \
--save_steps 5 \
--run_name quick_test_gee \
--wandb_project one-shot-gee
if [ $? -eq 0 ]; then
echo "✓ 快速训练测试通过"
else
echo "✗ 快速训练测试失败"
exit 1
fi
echo ""
echo "所有快速测试通过！✓"
echo "现在可以运行完整的训练和评估脚本"
|
