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?

[ujaejoon]

try followings:

1. fix MLP in context_cluster.py

2. fix Cluster in context_cluster.py