From 69b585a9925ae8a0da95adba13799a3163c299a8 Mon Sep 17 00:00:00 2001
From: sayakpaul <spsayakpaul@gmail.com>
Date: Wed, 25 Dec 2024 11:35:09 +0530
Subject: [PATCH] fixes

---
 finetrainers/utils/checkpointing.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/finetrainers/utils/checkpointing.py b/finetrainers/utils/checkpointing.py
index 11b94fc..c800c47 100644
--- a/finetrainers/utils/checkpointing.py
+++ b/finetrainers/utils/checkpointing.py
@@ -31,6 +31,8 @@ def get_latest_ckpt_path_to_resume_from(
             )
             resume_from_checkpoint = None
             initial_global_step = 0
+            global_step = 0
+            first_epoch = 0
             resume_from_checkpoint_path = None
         else:
             logger.info(f"Resuming from checkpoint {path}")