From b7d24978a4e2811f65f3a1cdd00cd0999487c81d Mon Sep 17 00:00:00 2001 From: Joshua David Date: Sat, 6 Jul 2024 22:46:32 -0700 Subject: [PATCH] Begin implementing a training edge case to pickup where the training left off if interrupeted. --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index fdbfba1..6f8ed26 100644 --- a/train.py +++ b/train.py @@ -177,7 +177,7 @@ def train( f"Resumed training from {resume_from_checkpoint} at epoch {start_epoch}" ) - for epoch in range(epochs): + for epoch in range(start_epoch, epochs): model.train() total_loss = 0