From acbf84f353804bca9438cd30fee63058e8ba5b56 Mon Sep 17 00:00:00 2001 From: wangmengke Date: Thu, 9 Nov 2023 18:29:26 +0800 Subject: [PATCH] change train script --- .github/workflows/pr_merged.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/pr_merged.yaml b/.github/workflows/pr_merged.yaml index 0acabb4..9a2e7b2 100644 --- a/.github/workflows/pr_merged.yaml +++ b/.github/workflows/pr_merged.yaml @@ -49,4 +49,4 @@ jobs: run: | source /mnt/petrelfs/share_data/llm_env/env/llm-flash2.0 export PYTHONPATH=$PWD:$PYTHONPATH - srun -p ${SLURM_PARTITION} --job-name=internlm-acc-test-${GITHUB_RUN_ID}-${GITHUB_JOB} --quotatype=spot -N 1 -n 8 --ntasks-per-node=8 --gpus-per-task=1 python ./train.py --config ./tests/test_training/7B_check_acc.py + srun -p ${SLURM_PARTITION} --job-name=internlm-acc-test-${GITHUB_RUN_ID}-${GITHUB_JOB} --quotatype=spot -N 1 -n 8 --ntasks-per-node=8 --gpus-per-task=1 python ./tests/test_training/train_CI.py --config ./tests/test_training/7B_check_acc.py