From 55ef29df8039f7199a033da5a29b8e825964ecc9 Mon Sep 17 00:00:00 2001 From: lijiaxing Date: Fri, 15 Dec 2023 18:29:04 +0800 Subject: [PATCH] auto save --- internlm/train/training_internlm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/internlm/train/training_internlm.py b/internlm/train/training_internlm.py index c190fa6..0fec19e 100644 --- a/internlm/train/training_internlm.py +++ b/internlm/train/training_internlm.py @@ -455,8 +455,8 @@ def record_current_batch_training_metrics( save_frequency = int(save_frequency[0]) # assign save_frequency - # when the "checkpoint_every" is "auto", no snapshot will be implemented - # when the "save_frequency" is less than the "checkpoint_every" passed in, 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 performed if gpc.config.ckpt.checkpoint_every == "auto": gpc.config.ckpt.checkpoint_every = save_frequency else: