WongKinYiu / yolov9

Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information
GNU General Public License v3.0
8.55k stars 1.3k forks source link

Error pretrained net yolov9 #163

Open unrue opened 4 months ago

unrue commented 4 months ago

I successfully trained ayolov9 from scratch, and now I would try a pretrained net as follow:

python -m torch.distributed.run --nnodes $SLURM_NNODES --node_rank $SLURM_PROCID --master_addr $MASTER_ADDR --master_port $MASTER_PORT --nproc_per_node 4 train_dual.py --workers 4 --sync-bn --batch 128 --data my_dataset.yaml --img 640 --cfg models/detect/yolov9-c.yaml --weights modelli/yolov9c.pt --name yolov9-c_pretrained_bc --hyp hyp.scratch-high.yaml --min-items 0 --epochs 500 --close-mosaic 15 --device 0,1,2,3 --epochs 500

having the error:

AttributeError: Can't get attribute 'RepNCSPELAN4' on <module 'ultralytics.nn.modules.block' from '/leonardo_scratch/large/userinternal/gfatigat/Deep_Learning/MIC_DL/detectron2_alternative/yolov8_env/lib/python3.10/site-packages/ultralytics/nn/modules/block.py'>

How can I train starting from a pretrained net? Thanks.

svetadomoi commented 4 months ago

Try to update the ultralytics package via pip install ultralytics -U

WongKinYiu commented 4 months ago

yolov9c.pt is converted into yolov8 format. please download yolov9-c.pt from this repo.