From 591ad3e0ed0145d14561aa6506f4c3ceec84dad8 Mon Sep 17 00:00:00 2001 From: Patrick Miles Date: Wed, 27 May 2026 08:58:09 -0700 Subject: [PATCH] fix restart epoch bug in trainer --- ScaFFold/utils/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()