github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix

pull/577/head
877825076@qq.com 2024-01-10 15:50:17 +08:00
parent d2e01dfed8
commit e41a6b7c28
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internlm/utils/model_checkpoint.py
View File

@ -996,7 +996,7 @@ now step_count is {train_state.step_count}",
if ( if (
train_state.step_count > 0 train_state.step_count > 0
and train_state.step_count % self.checkpoint_every == 0 and train_state.step_count % self.checkpoint_every == 0
or train_state.step_count == train_state.total_steps or train_state.batch_count == (train_state.total_steps - 1)
): ):
save_ckpts, save_type = True, CheckpointSaveType.NORMAL_CHECKPOINT save_ckpts, save_type = True, CheckpointSaveType.NORMAL_CHECKPOINT