THU-MIG / yolov10

YOLOv10: Real-Time End-to-End Object Detection [NeurIPS 2024]
https://arxiv.org/abs/2405.14458
GNU Affero General Public License v3.0
9.99k stars 990 forks source link

training script is downloading yolov8n.pt #389

Open mzhadigerov opened 3 months ago

mzhadigerov commented 3 months ago

I am following this tutorial to train a model on custom dataset.

I downloaded weights yolov10m.pt, and mentioned it like:

yolo task=detect mode=train epochs=250 batch=64 plots=True \
> model=weights/yolov10m.pt \
> data=datasets/data.yaml

but it downloads yolov8n.pt for some reason. Why?

bingYes520 commented 3 months ago

地址问题

mzhadigerov commented 3 months ago

@bingYes520

project_directory/
│
├── weights/
│   └── yolov10m.pt
│
└── datasets/
    ├── test/
    ├── train/
    ├── valid/
    └── data.yaml
bingYes520 commented 3 months ago

The issue with yolov8n.pt might be due to the use of Automatic Mixed Precision (AMP). You can either download yolov8n.pt to use mixed precision or set amp to False to disable mixed precision, in which case you won't need to download yolov8n.pt.