error in MMD_distance, The size of tensor a (20) must match the size of tensor b (256) at non-singleton dimension 3

Question

error in MMD_distance, The size of tensor a (20) must match the size of tensor b (256) at non-singleton dimension 3

saranadeem4585 opened this issue 6 months ago · comments

Hi,
Hope you are doing well.
I read your paper and it is amazing and now I am running it on a custom dataset. but it is giving me this error.
Please help me resolve this issue.
Thanks!

saranadeem4585 · Answer 1 · Mon Jan 29 2024 23:10:49 GMT+0800 (China Standard Time)

256 features, because of yolov5n.

Yipeng Gao · Answer 2 · Wed Jan 31 2024 23:31:58 GMT+0800 (China Standard Time)

Hi! Sorry for the late reply due to the CVPR rebuttal. Could you please provide the error with log? I think a potential modification exists in the Ranker (nn.Model) that needs to specify the output dimension. In this codebase, I set the number corresponding to YOLOv5 X by default.