WongKinYiu / yolov9

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

How to use validation score in training yolov9 model #486

Open KyriakosChris opened 4 months ago

KyriakosChris commented 4 months ago

I am training a yolov9 tiny model, and i tried to uncomment the lines 196 and 197 in the val_dual.py, and got this error in the line 197. Do you have any idea how to use the val in training, leaving them in comment it works, but the validation is always zero, i wanted to have validation during my training

RuntimeError: shape '[66, 66, -1]' is invalid for input of size 266112

babakbch commented 3 months ago

this is an ongoing issue apparently, I have the same problem but so far no solution