yolov5 icon indicating copy to clipboard operation
yolov5 copied to clipboard

Fix bug with resume

Open savchenkoyana opened this issue 1 year ago • 0 comments

Hello, I noticed strange code behavior.

I ran training on coco128:

python train.py --name exp_coco128

trained for 2 epochs and then stopped.

Then ran:

python train.py --weights runs/train/exp_coco128/weights/last.pt

And the training continued from the 3rd epoch, although I didn't use --resume flag. What's the point of --resume command if I resume my previous training without using it anyways?

I suggest calling smart_resume function only if resume == True. What do you think?

Thank you!

savchenkoyana avatar Aug 09 '22 10:08 savchenkoyana

@savchenkoyana PR is merged. Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐

glenn-jocher avatar Aug 16 '22 22:08 glenn-jocher