search

HZAI-ZJNU / Mamba-YOLO

the official pytorch implementation of “Mamba-YOLO：SSMs-based for Object Detection”
Apache License 2.0
227 stars 27 forks source link

train error #43

Open Laulen opened 3 weeks ago

Laulen commented 3 weeks ago

Has anyone encountered this problem? The first round of training revealed the following errors.

image
yazLL commented 1 day ago

I have exactly the same error, have you resolved it? Ubuntu 20.04 RTX3090 conda pytorch 2.3 cuda 11.8

python mbyolo_train.py --device 0 --batch 32 --epochs 100 --task train --data ultralytics/cfg/datasets/coco.yaml --config ultralytics/cfg/models/mamba-yolo/Mamba-YOLO-T.yaml --amp --project ./output_dir/mscoco --name mambayolo_n

changed sorce data to my path in coco.yaml path: /home/ml/250GB/Mamba-YOLO-main/datasets/MSCOCO2017 # dataset root dir others not been touched

[solved] wrong folder structure made it can't find labels cause validation error