runtime error

Exit code: 1. Reason: 4]) from checkpoint, the shape in current model is torch.Size([512]). size mismatch for decs.0.0.weight: copying a param with shape torch.Size([1024, 512, 4, 4]) from checkpoint, the shape in current model is torch.Size([512, 256, 4, 4]). size mismatch for decs.0.0.bias: copying a param with shape torch.Size([512]) from checkpoint, the shape in current model is torch.Size([256]). size mismatch for decs.0.1.weight: copying a param with shape torch.Size([512]) from checkpoint, the shape in current model is torch.Size([256]). size mismatch for decs.0.1.bias: copying a param with shape torch.Size([512]) from checkpoint, the shape in current model is torch.Size([256]). size mismatch for decs.1.0.weight: copying a param with shape torch.Size([1024, 256, 4, 4]) from checkpoint, the shape in current model is torch.Size([512, 128, 4, 4]). size mismatch for decs.1.0.bias: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([128]). size mismatch for decs.1.1.weight: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([128]). size mismatch for decs.1.1.bias: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([128]). size mismatch for decs.2.0.weight: copying a param with shape torch.Size([512, 128, 4, 4]) from checkpoint, the shape in current model is torch.Size([256, 64, 4, 4]). size mismatch for decs.2.0.bias: copying a param with shape torch.Size([128]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for decs.2.1.weight: copying a param with shape torch.Size([128]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for decs.2.1.bias: copying a param with shape torch.Size([128]) from checkpoint, the shape in current model is torch.Size([64]). size mismatch for final.weight: copying a param with shape torch.Size([256, 3, 4, 4]) from checkpoint, the shape in current model is torch.Size([128, 3, 4, 4]).

Container logs: