Please check the model.py and option.py version is same with .pth model
Numerous state_dict names and dimensions is not match!
-------------------------------------------- Wrong message ------------------------------------------------
RuntimeError: Error(s) in loading state_dict for CFNet:
size mismatch for srb_1.upBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.upBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.upBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.upBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.upBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.upBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.downBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.downBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.downBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.downBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.downBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_1.downBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for out_over.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.upBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.upBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.upBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.upBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.upBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.upBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.downBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.downBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.downBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.downBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.downBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for srb_2.downBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for out_under.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for out_1.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for out_1.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for out_1.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for out_2.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for out_2.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for out_2.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.upBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.upBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.upBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.upBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.upBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.upBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.downBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.downBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.downBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.downBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.downBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over0.downBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.upBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.upBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.upBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.upBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.upBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.upBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.downBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.downBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.downBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.downBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.downBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under0.downBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.upBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.upBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.upBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.upBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.upBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.upBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.downBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.downBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.downBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.downBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.downBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over1.downBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.upBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.upBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.upBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.upBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.upBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.upBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.downBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.downBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.downBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.downBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.downBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under1.downBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.upBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.upBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.upBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.upBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.upBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.upBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.downBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.downBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.downBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.downBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.downBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_over2.downBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.upBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.upBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.upBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.upBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.upBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.upBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.downBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.downBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.downBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.downBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.downBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).
size mismatch for cfb_under2.downBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 8, 8]) from checkpoint, the shape in current model is torch.Size([64, 64, 6, 6]).