From 2f6208c3560f4217f4c5d370939db44aeff41442 Mon Sep 17 00:00:00 2001
From: Thomas Friedel <thomas.friedel@gmail.com>
Date: Sun, 26 May 2024 04:32:05 +0200
Subject: [PATCH] fix for KeyError: 'epoch' (#48)

* fix for KeyError: 'epoch'
---
 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 3d16afea..cfdf80ed 100644
--- a/ultralytics/engine/trainer.py
+++ b/ultralytics/engine/trainer.py
@@ -659,7 +659,7 @@ class BaseTrainer:
 
     def resume_training(self, ckpt):
         """Resume YOLO training from given epoch and best fitness."""
-        if ckpt is None:
+        if ckpt is None or not self.resume:
             return
         best_fitness = 0.0
         start_epoch = ckpt["epoch"] + 1