diff --git a/ScaFFold/utils/trainer.py b/ScaFFold/utils/trainer.py
index 3add912..33cc30c 100644
--- a/ScaFFold/utils/trainer.py
+++ b/ScaFFold/utils/trainer.py
@@ -568,7 +568,7 @@ def train(self):
         Execute model training
         """
 
-        epoch = 1
+        epoch = self.start_epoch
         dice_score_train = 0
         with open(self.outfile_path, "a", newline="") as outfile:
             start = time.time()