yolov5
yolov5 copied to clipboard
Fix bug with resume
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 PR is merged. Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐