alibaba / alibaba/esod

Error with training using Yolov8m

Open
#18 2 comments 0 reactions 0 assignees View on GitHub
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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.