Tensor not match - Githubissues

pooya-mohammadi / yolov5-gradcam

Visualizing Yolov5's layers using GradCam

MIT License

281 stars 47 forks source link

Tensor not match #28

Open frisk99 opened 1 year ago

frisk99 commented 1 year ago

hello, I trianed it on DOTA datasets,but get error. RuntimeError: The size of tensor a (80) must match the size of tensor b (1280) at non-singleton dimension 1

NusratTahira commented 2 months ago

Hi, I also got the same problem. Did you fix it? RuntimeError: The size of tensor a (32) must match the size of tensor b (512) at non-singleton dimension 1