can't load detection pretrained model

[Missyfirst]

excuse me,there is a error of detection pretrained model, raise error:
size mismatch for backbone.network.0.0.mlp.fc1.weight: copying a param with shape torch.Size([512, 64, 1, 1]) from checkpoint, the
shape in current model is torch.Size([512, 64]).
size mismatch for backbone.network.0.0.mlp.fc2.weight: copying a param with shape torch.Size([64, 512, 1, 1]) from checkpoint, the
shape in current model is torch.Size([64, 512]).
Hope you can provide new checkpoints. thank you.

[sazed49]

I am also having the same issue. Have you solved it?

[Jaejoon Yoo]

try followings:

1. fix MLP in context_cluster.py
   

2. fix Cluster in context_cluster.py