模型训练问题

[seven48264]

你好，我想自己重新在自己的数据集上训练这个模型，请问有相应的训练代码嘛

[homesomele]

你好，请问你保存了模型之后测试有没有遇到加载失败的问题呀

RuntimeError: Error(s) in loading state_dict for SwinJSCC:
size mismatch for encoder.layers.0.blocks.1.attn_mask: copying a param with shape torch.Size([1536, 64, 64]) from checkpoint, the shape in current model is torch.Size([256, 64, 64]).
size mismatch for encoder.layers.1.blocks.1.attn_mask: copying a param with shape torch.Size([384, 64, 64]) from checkpoint, the shape in current model is torch.Size([64, 64, 64]).
size mismatch for encoder.layers.2.blocks.1.attn_mask: copying a param with shape torch.Size([96, 64, 64]) from checkpoint, the shape in current model is torch.Size([16, 64, 64]).
size mismatch for encoder.layers.2.blocks.3.attn_mask: copying a param with shape torch.Size([96, 64, 64]) from checkpoint, the shape in current model is torch.Size([16, 64, 64]).
size mismatch for encoder.layers.2.blocks.5.attn_mask: copying a param with shape torch.Size([96, 64, 64]) from checkpoint, the shape in current model is torch.Size([16, 64, 64]).
size mismatch for encoder.layers.3.blocks.1.attn_mask: copying a param with shape torch.Size([24, 64, 64]) from checkpoint, the shape in current model is torch.Size([4, 64, 64]).
size mismatch for decoder.layers.0.blocks.1.attn_mask: copying a param with shape torch.Size([24, 64, 64]) from checkpoint, the shape in current model is torch.Size([4, 64, 64]).
size mismatch for decoder.layers.1.blocks.1.attn_mask: copying a param with shape torch.Size([96, 64, 64]) from checkpoint, the shape in current model is torch.Size([16, 64, 64]).
size mismatch for decoder.layers.1.blocks.3.attn_mask: copying a param with shape torch.Size([96, 64, 64]) from checkpoint, the shape in current model is torch.Size([16, 64, 64]).
size mismatch for decoder.layers.1.blocks.5.attn_mask: copying a param with shape torch.Size([96, 64, 64]) from checkpoint, the shape in current model is torch.Size([16, 64, 64]).
size mismatch for decoder.layers.2.blocks.1.attn_mask: copying a param with shape torch.Size([384, 64, 64]) from checkpoint, the shape in current model is torch.Size([64, 64, 64]).
size mismatch for decoder.layers.3.blocks.1.attn_mask: copying a param with shape torch.Size([1536, 64, 64]) from checkpoint, the shape in current model is torch.Size([256, 64, 64]).