how to set MoCoViT-0.5?

Question

how to set MoCoViT-0.5?

Opened this issue 2 years ago · 1 comments

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

Answer 1 · 2022-08-23T14:41:37.000Z

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.