-
-
Notifications
You must be signed in to change notification settings - Fork 17.2k
Closed
Labels
bugSomething isn't workingSomething isn't working
Description
Hi, thanks for this amazing repo first of all!
I am working with the v2.0 release of the repo and so far I got quite promising results on my custom dataset. I tried to use --evolve today but I am getting a TypeError from the print_mutation function in utils.py at the end of the first run. It seems to be related to 'SGD' in hyp dictionary - which is a string. I see that the optimizer parameter was added to the dict in v2.0 so I guess this wasn't an issue in the previous release.
Input:
!python yolov5/train.py --img 512 --batch 16 --epochs 100 --yolov5/data/custom_data.yaml \
--cfg yolov5/models/custom_yolov5s.yaml \
--weights yolov5/weights/yolov5s.pt --cache --evolve
Output:
Traceback (most recent call last):
File "yolov5/train.py", line 516, in <module>
print_mutation(hyp, results, opt.bucket)
File "/content/drive/.shortcut-targets-by-id/1hQ281-8ogL2dJzHbXZSWEYDIYei6RRbe/YOLOv5/yolov5/utils/utils.py", line 830, in print_mutation
b = '%10.3g' * len(hyp) % tuple(hyp.values()) # hyperparam values
TypeError: must be real number, not str
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working