xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix for KeyError: 'epoch' (#48)

Browse Source 
* fix for KeyError: 'epoch'
This commit is contained in:
Thomas Friedel 2024-05-26 04:32:05 +02:00 committed by GitHub
parent f19f3e521f
commit 2f6208c356
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/engine/trainer.py
View File

@ -659,7 +659,7 @@ class BaseTrainer:
def resume_training(self, ckpt): def resume_training(self, ckpt):
"""Resume YOLO training from given epoch and best fitness.""" """Resume YOLO training from given epoch and best fitness."""
if ckpt is None: if ckpt is None or not self.resume:
return return
best_fitness = 0.0 best_fitness = 0.0
start_epoch = ckpt["epoch"] + 1 start_epoch = ckpt["epoch"] + 1