Skip to content

Commit

Permalink
test(case): add load ckpt then assert loss into ci (#29)
Browse files Browse the repository at this point in the history
Co-authored-by: MK1512 <wangmengke@pjlab.org.cn>
  • Loading branch information
kkscilife and MK1512 authored Jan 26, 2024
1 parent b7089a9 commit 4b198c2
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions .github/workflows/weekly_test.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -113,3 +113,16 @@ jobs:
run: |
source $evo_env
srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} -N 1 -n 1 --gres=gpu:8 python -m pytest -s ./tests/test_model/test_norm.py
load_ckpt_then_assert_loss:
runs-on: [t_cluster]
timeout-minutes: 10
steps:
- uses: actions/checkout@v3
with:
ref: ${{ github.event_name == 'schedule' && 'develop' || github.event_name == 'workflow_dispatch' && '' }}

- name: test_ckpt_loss
run: |
source $evo_env
srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} -N 1 -n 1 --gres=gpu:8 python -m pytest -s ./tests/test_training/test_load_ckpt_loss.py

0 comments on commit 4b198c2

Please sign in to comment.