When running test.py, loading the pretrained model of Rain200H causes an error.

[fashyon]

Unexpected key(s) in state_dict: "encoder_level1.0.attn.ta.sharedMLP.0.weight", "encoder_level1.0.attn.ta.sharedMLP.2.weight", "encoder_level1.0.attn.ca.sharedMLP.0.weight", "encoder_level1.0.attn.ca.sharedMLP.2.weight", "encoder_level1.0.attn.sa.conv.weight", "encoder_level1.1.attn.ta.sharedMLP.0.weight", "encoder_level1.1.attn.ta.sharedMLP.2.weight", "encoder_level1.1.attn.ca.sharedMLP.0.weight", "encoder_level1.1.attn.ca.sharedMLP.2.weight", "encoder_level1.1.attn.sa.conv.weight", "encoder_level1.2.attn.ta.sharedMLP.0.weight", "encoder_level1.2.attn.ta.sharedMLP.2.weight", "encoder_level1.2.attn.ca.sharedMLP.0.weight", "encoder_level1.2.attn.ca.sharedMLP.2.weight", "encoder_level1.2.attn.sa.conv.weight", "encoder_level1.3.attn.ta.sharedMLP.0.weight", "encoder_level1.3.attn.ta.sharedMLP.2.weight", "encoder_level1.3.attn.ca.sharedMLP.0.weight", "encoder_level1.3.attn.ca.sharedMLP.2.weight", "encoder_level1.3.attn.sa.conv.weight", "encoder_level2.0.attn.ta.sharedMLP.0.weight", "encoder_level2.0.attn.ta.sharedMLP.2.weight", "encoder_level2.0.attn.ca.sharedMLP.0.weight", "encoder_level2.0.attn.ca.sharedMLP.2.weight", "encoder_level2.0.attn.sa.conv.weight", "encoder_level2.1.attn.ta.sharedMLP.0.weight", "encoder_level2.1.attn.ta.sharedMLP.2.weight", "encoder_level2.1.attn.ca.sharedMLP.0.weight", "encoder_level2.1.attn.ca.sharedMLP.2.weight", "encoder_level2.1.attn.sa.conv.weight", "encoder_level2.2.attn.ta.sharedMLP.0.weight", "encoder_level2.2.attn.ta.sharedMLP.2.weight", "encoder_level2.2.attn.ca.sharedMLP.0.weight", "encoder_level2.2.attn.ca.sharedMLP.2.weight", "encoder_level2.2.attn.sa.conv.weight", "encoder_level2.3.attn.ta.sharedMLP.0.weight", "encoder_level2.3.attn.ta.sharedMLP.2.weight", "encoder_level2.3.attn.ca.sharedMLP.0.weight", "encoder_level2.3.attn.ca.sharedMLP.2.weight", "encoder_level2.3.attn.sa.conv.weight", "encoder_level3.0.attn.ta.sharedMLP.0.weight", "encoder_level3.0.attn.ta.sharedMLP.2.weight", "encoder_level3.0.attn.ca.sharedMLP.0.weight", "encoder_level3.0.attn.ca.sharedMLP.2.weight", "encoder_level3.0.attn.sa.conv.weight", "encoder_level3.1.attn.ta.sharedMLP.0.weight", "encoder_level3.1.attn.ta.sharedMLP.2.weight", "encoder_level3.1.attn.ca.sharedMLP.0.weight", "encoder_level3.1.attn.ca.sharedMLP.2.weight", "encoder_level3.1.attn.sa.conv.weight", "encoder_level3.2.attn.ta.sharedMLP.0.weight", "encoder_level3.2.attn.ta.sharedMLP.2.weight", "encoder_level3.2.attn.ca.sharedMLP.0.weight", "encoder_level3.2.attn.ca.sharedMLP.2.weight", "encoder_level3.2.attn.sa.conv.weight", "encoder_level3.3.attn.ta.sharedMLP.0.weight", "encoder_level3.3.attn.ta.sharedMLP.2.weight", "encoder_level3.3.attn.ca.sharedMLP.0.weight", "encoder_level3.3.attn.ca.sharedMLP.2.weight", "encoder_level3.3.attn.sa.conv.weight", "encoder_level3.4.attn.ta.sharedMLP.0.weight", "encoder_level3.4.attn.ta.sharedMLP.2.weight", "encoder_level3.4.attn.ca.sharedMLP.0.weight", "encoder_level3.4.attn.ca.sharedMLP.2.weight", "encoder_level3.4.attn.sa.conv.weight", "encoder_level3.5.attn.ta.sharedMLP.0.weight", "encoder_level3.5.attn.ta.sharedMLP.2.weight", "encoder_level3.5.attn.ca.sharedMLP.0.weight", "encoder_level3.5.attn.ca.sharedMLP.2.weight", "encoder_level3.5.attn.sa.conv.weight", "encoder_level3.6.attn.ta.sharedMLP.0.weight", "encoder_level3.6.attn.ta.sharedMLP.2.weight", "encoder_level3.6.attn.ca.sharedMLP.0.weight", "encoder_level3.6.attn.ca.sharedMLP.2.weight", "encoder_level3.6.attn.sa.conv.weight", "encoder_level3.7.attn.ta.sharedMLP.0.weight", "encoder_level3.7.attn.ta.sharedMLP.2.weight", "encoder_level3.7.attn.ca.sharedMLP.0.weight", "encoder_level3.7.attn.ca.sharedMLP.2.weight", "encoder_level3.7.attn.sa.conv.weight", "decoder_level3.0.attn.ta.sharedMLP.0.weight", "decoder_level3.0.attn.ta.sharedMLP.2.weight", "decoder_level3.0.attn.ca.sharedMLP.0.weight", "decoder_level3.0.attn.ca.sharedMLP.2.weight", "decoder_level3.0.attn.sa.conv.weight", "decoder_level3.1.attn.ta.sharedMLP.0.weight", "decoder_level3.1.attn.ta.sharedMLP.2.weight", "decoder_level3.1.attn.ca.sharedMLP.0.weight", "decoder_level3.1.attn.ca.sharedMLP.2.weight", "decoder_level3.1.attn.sa.conv.weight", "decoder_level2.0.attn.ta.sharedMLP.0.weight", "decoder_level2.0.attn.ta.sharedMLP.2.weight", "decoder_level2.0.attn.ca.sharedMLP.0.weight", "decoder_level2.0.attn.ca.sharedMLP.2.weight", "decoder_level2.0.attn.sa.conv.weight", "decoder_level2.1.attn.ta.sharedMLP.0.weight", "decoder_level2.1.attn.ta.sharedMLP.2.weight", "decoder_level2.1.attn.ca.sharedMLP.0.weight", "decoder_level2.1.attn.ca.sharedMLP.2.weight", "decoder_level2.1.attn.sa.conv.weight", "decoder_level1.0.attn.ta.sharedMLP.0.weight", "decoder_level1.0.attn.ta.sharedMLP.2.weight", "decoder_level1.0.attn.ca.sharedMLP.0.weight", "decoder_level1.0.attn.ca.sharedMLP.2.weight", "decoder_level1.0.attn.sa.conv.weight", "decoder_level1.1.attn.ta.sharedMLP.0.weight", "decoder_level1.1.attn.ta.sharedMLP.2.weight", "decoder_level1.1.attn.ca.sharedMLP.0.weight", "decoder_level1.1.attn.ca.sharedMLP.2.weight", "decoder_level1.1.attn.sa.conv.weight".

[hmfjx123]

Same problem when running test.py for Rain200L

[bee3131263]

Same problem here