how to set MoCoViT-0.5?
Opened this issue · 1 comments
winterxx commented
when set width_mult=0.5, i have a error:
File "/xx/MoCoViT-PyTorch/mocovit.py", line 74, in forward
return self.ffn(x) + x
RuntimeError: The size of tensor a (80) must match the size of tensor b (160) at non-singleton dimension 1
smitheric95 commented
Thank you for posting this issue. The underlying problem is that the output of the backbone GhostNet network is 160 channels, and therefore doesn't match up well outside of the default parameters of MoCoViT.
I will try to find an elegant solution to this.