RuntimeError: Error(s) in loading state_dict for DataParallel:
size mismatch for module.block.upBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.upBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.upBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.upBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.upBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.upBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.downBlocks.0.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.downBlocks.1.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.downBlocks.2.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.downBlocks.3.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.downBlocks.4.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.block.downBlocks.5.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).
size mismatch for module.out.0.weight: copying a param with shape torch.Size([64, 64, 6, 6]) from checkpoint, the shape in current model is torch.Size([64, 64, 8, 8]).