diff --git a/.cnb.yml b/.cnb.yml index 18912ae..4101d74 100644 --- a/.cnb.yml +++ b/.cnb.yml @@ -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 \ No newline at end of file + 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 "=== 训练完成 ===" \ No newline at end of file