xiaomabufei / FGAHOI

29 stars 4 forks source link

Evaluation with Swin-large ckpt came across loading ckpt error. Can you provide the script to evaluate with Swin-large? #15

Open shaniaos opened 11 months ago

shaniaos commented 11 months ago

Hi, Thanks for your great work! I'm very impressed by the strong performance of your method and trying to reproduce the result on Swin-large. However, as only the swin-tiny evaluation script is provided and not the swin-large, I tried to run with swin-large ckpt by replacing "--backbone swin_tiny" with "--backbone swin_large". However, I came across such error:

        size mismatch for backbone.0.body.layers.2.blocks.4.attn.relative_position_index: copying a param with shape torch.Size([144, 144]) from checkpoint, the shape in current model is torch.Size([49, 49]).

How can I fix this and run with your swin-large ckpt? Can you provide the evaluation script for swin-large? I'm looking forward to your reply. Thanks a lot!