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

auto save

pull/546/head
lijiaxing 2023-12-15 18:29:04 +08:00
parent 81f51fd0ff
commit 55ef29df80
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internlm/train/training_internlm.py
View File

@ -455,8 +455,8 @@ def record_current_batch_training_metrics(
save_frequency = int(save_frequency[0]) save_frequency = int(save_frequency[0])
# assign save_frequency # assign save_frequency
# when the "checkpoint_every" is "auto", no snapshot will be implemented # when the "checkpoint_every" is "auto", no snapshot will be performed
# when the "save_frequency" is less than the "checkpoint_every" passed in, no snapshot will be implemented # when the "save_frequency" is less than the "checkpoint_every" passed in, no snapshot will be performed
if gpc.config.ckpt.checkpoint_every == "auto": if gpc.config.ckpt.checkpoint_every == "auto":
gpc.config.ckpt.checkpoint_every = save_frequency gpc.config.ckpt.checkpoint_every = save_frequency
else: else: