1 files changed, 45 insertions, 0 deletions
diff --git a/scripts/quick_test_gee.sh b/scripts/quick_test_gee.sh
new file mode 100755
index 0000000..43763df
--- /dev/null
+++ b/scripts/quick_test_gee.sh
@@ -0,0 +1,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 "现在可以运行完整的训练和评估脚本" 
+\ No newline at end of file