This commit is contained in:
sayakpaul
2024-12-25 11:35:09 +05:30
parent 875cb72f63
commit 69b585a992
+2
View File
@@ -31,6 +31,8 @@ def get_latest_ckpt_path_to_resume_from(
) )
resume_from_checkpoint = None resume_from_checkpoint = None
initial_global_step = 0 initial_global_step = 0
global_step = 0
first_epoch = 0
resume_from_checkpoint_path = None resume_from_checkpoint_path = None
else: else:
logger.info(f"Resuming from checkpoint {path}") logger.info(f"Resuming from checkpoint {path}")