FFC
FFC copied to clipboard
Pretrained models doesn't match code
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?
Hi!
I fix it with this code:
I hope it can help you.
In my case, the name of ckpt layers don't matched with that from model, I fix it with this code: