chore: add debug echo to pipeline script

This commit is contained in:
shuiyu1123 committed 2026-06-18 19:06:17 +00:00
1 parent ffb5825407
commit 5732229ed5
1 file changed
+11 -2
+11 -2
View File
@@ -9,8 +9,17 @@ master:
env:
CUDA_VISIBLE_DEVICES: "0"
PYTHONUNBUFFERED: "1"
git:
lfs: false
stages:
- name: train
script: |
cd /workspace && git clone https://cnb.cool/ETeam_sy/xiaoli-llm-chat.git
cd xiaoli-llm-chat && python3 scripts/train_cnb.py --epochs 100 --batch_size 16
set -e
echo "=== 开始训练 ==="
cd /workspace
ls -la
cd /workspace/xiaoli-llm-chat
nvidia-smi
echo "=== 启动训练 ==="
python3 scripts/train_cnb.py --epochs 100 --batch_size 16
echo "=== 训练完成 ==="