-
Notifications
You must be signed in to change notification settings - Fork 16
Description
Hi! I wanted to train your algorithm using the yolov8m model. I used the downloaded weights as mentioned in the readme and the training command, only changing the model type into yolov8. I used the following command:
DATASET=visdrone MODEL=yolov8m GPUS=0 BATCH_SIZE=4 IMAGE_SIZE=1536 EPOCHS=10 bash ./scripts/train.sh
And got this error:
self.m = nn.ModuleList(Bottleneck(self.c, self.c, shortcut, g, k=((3, 3), (3, 3)), e=1.0) for _ in range(n)) TypeError: __init__() got an unexpected keyword argument 'k'
The error was in a file ./models/common.py.
I think this is a problem with mismatch of the yaml file and downloaded weights, since your code supports only a yolov5 architecture.
I will be happy with any advice you could give. Thanks in advance!