runtime error

Exit code: 1. Reason: e([64, 256, 2, 3]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for model.model.blocks.mask_decoder.mask_conv.1.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([1, 64, 1, 1]). size mismatch for model.model.blocks.mask_decoder.mask_conv.1.bias: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([1]). size mismatch for model.model.blocks.mask_decoder.mask_conv.2.weight: copying a param with shape torch.Size([64]) from checkpoint, the shape in current model is torch.Size([1]). size mismatch for model.model.blocks.mask_decoder.mask_conv.3.weight: copying a param with shape torch.Size([1, 64, 1, 2]) from checkpoint, the shape in current model is torch.Size([1]). size mismatch for model.model.blocks.phase_decoder.dense_block.dense_block.0.1.weight: copying a param with shape torch.Size([64, 64, 2, 3]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for model.model.blocks.phase_decoder.dense_block.dense_block.1.1.weight: copying a param with shape torch.Size([64, 128, 2, 3]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for model.model.blocks.phase_decoder.dense_block.dense_block.2.1.weight: copying a param with shape torch.Size([64, 192, 2, 3]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for model.model.blocks.phase_decoder.dense_block.dense_block.3.1.weight: copying a param with shape torch.Size([64, 256, 2, 3]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for model.model.blocks.phase_decoder.phase_conv_r.weight: copying a param with shape torch.Size([1, 64, 1, 2]) from checkpoint, the shape in current model is torch.Size([1, 64, 1, 1]). size mismatch for model.model.blocks.phase_decoder.phase_conv_i.weight: copying a param with shape torch.Size([1, 64, 1, 2]) from checkpoint, the shape in current model is torch.Size([1, 64, 1, 1]).

Container logs: