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

[saranadeem4585]

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]

256 features, because of yolov5n.

[Hlings]

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.