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

Hlings / AsyFOD

(CVPR2023) The PyTorch implementation of the "AsyFOD: An Asymmetric Adaptation Paradigm for Few-Shot Domain Adaptive Object Detection".

MIT License

33 stars 3 forks source link

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

Closed saranadeem4585 closed 2 months ago

saranadeem4585 commented 7 months ago

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 commented 7 months ago

256 features, because of yolov5n.

Hlings commented 7 months ago

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.