Hi,
I’m a 4th year CS student and as part of my engineering thesis I’m trying to recreate YOLO algorithm for object detection.
Despite making sure that both targets and inputs are converted to float and device before training I seem to experience this type of error and can’t solve it on my own.
The full source code is available in my public github repository 1
I believe that the problem may be relevant to the advanced loss function, but my debugging hasn’t helped.
Maybe someone could give me some tips as to how could I solve this problem?