Pretrained models doesn't match code

[dreminm]

Hi! Greatly appreciate your work! It seems that pretrained weights that you provided in issue #1 doesn't match the code:

• layers are named differently
• there is size mismatch in some layers, f.e. size mismatch for layer3.3.conv2.ffc.convl2l.weight: copying a param with shape torch.Size([192, 192, 3, 3]) from checkpoint, the shape in current model is torch.Size([128, 128, 3, 3]).

Can you please check it out?

[VuHuy-cse-9]

Hi!
I fix it with this code:

I hope it can help you.

[VuHuy-cse-9]

In my case, the name of ckpt layers don't matched with that from model, I fix it with this code: