From db42d180d90c9c1539d9c318b26a66434409bf34 Mon Sep 17 00:00:00 2001 From: Burhan <62214284+Burhan-Q@users.noreply.github.com> Date: Thu, 25 Jul 2024 17:49:04 -0400 Subject: [PATCH] Updates `save_period` to include first epoch (#14700) Co-authored-by: Glenn Jocher --- ultralytics/engine/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/trainer.py b/ultralytics/engine/trainer.py index 71840822..9d71810b 100644 --- a/ultralytics/engine/trainer.py +++ b/ultralytics/engine/trainer.py @@ -507,7 +507,7 @@ class BaseTrainer: self.last.write_bytes(serialized_ckpt) # save last.pt if self.best_fitness == self.fitness: self.best.write_bytes(serialized_ckpt) # save best.pt - if (self.save_period > 0) and (self.epoch > 0) and (self.epoch % self.save_period == 0): + if (self.save_period > 0) and (self.epoch >= 0) and (self.epoch % self.save_period == 0): (self.wdir / f"epoch{self.epoch}.pt").write_bytes(serialized_ckpt) # save epoch, i.e. 'epoch3.pt' def get_dataset(self):