pretrained model giving mismatch issue #23
Description
RuntimeError: Error(s) in loading state_dict for Generic_UNet:
size mismatch for conv_blocks_context.0.blocks.0.conv.weight: copying a param with shape torch.Size([30, 15, 3, 3]) from checkpoint, the shape in current model is torch.Size([30, 1, 3, 3]).