Giters

pooya-mohammadi / yolov5-gradcam

Visualizing Yolov5's layers using GradCam

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Tensor not match

frisk99 opened this issue · comments

commented

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