Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

yolov8n.yaml keeps getting downloaded regradless of the given model #6

Open
shayandaneshvar opened this issue Mar 22, 2023 · 1 comment

Comments

@shayandaneshvar
Copy link

shayandaneshvar commented Mar 22, 2023

Hi,
I use the following command to train a custom dataset on the pre-trained Yolov8 with yolov8s weights, but the yolov8n is downloaded every time and I am guessing that the input for the model is being ignored.
Am I doing it wrong?

yolo detect train data={DATASET_YML} model=yolov8s.yaml pretrained=True epochs={epochs} imgsz={image_size} verbose=True

I also wanted to know what parameters should I set, so that I can train yolov8s similarly to yolov5s so that the same IoU and augmentation are used.

Also, what should I set to make the model run on GPU? The default value for the device param is null but it is missing from the document what will be chosen in the default mode.

@glenn-jocher
Copy link
Member

@shayandaneshvar for YOLOv8-related issues please raise an issue on https://github.com/ultralytics/ultralytics

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants