Error with training using Yolov8m
- Dominant language
- Python
- Stars
- 140
- Forks
- 21
- PR merge metrics
- No merged PRs in 30d
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!
Contributor guide
No contributing guide indexed for this repository
Research direction
Start by reproducing the command from the issue and inspect the model construction in models/common.py, then compare it with the yolov8m weights and the architecture described in the README. Done means the documented training command no longer raises the unexpected keyword argument 'k' error, or the supported model and weight combinations are clearly documented.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- computer-vision, machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100