MockingBird 模型加载错误，维度错误

模型加载错误，维度错误

Open wangsheng0131 opened this issue 1 year ago • 6 comments

使用的最新的main分支代码，合成模型用的是https://github.com/babysor/MockingBird/wiki/Quick-Start-(Newbie) wiki中的模型，https://github.com/babysor/MockingBird/issues/37 中的方案，并解决不了问题。

报错如下：

RuntimeError: Error(s) in loading state_dict for Tacotron: size mismatch for encoder.embedding.weight: copying a param with shape torch.Size([70, 512]) from checkpoint, the shape in current model is torch.Size([75, 512]). size mismatch for encoder_proj.weight: copying a param with shape torch.Size([128, 512]) from checkpoint, the shape in current model is torch.Size([128, 1024]). size mismatch for decoder.attn_rnn.weight_ih: copying a param with shape torch.Size([384, 768]) from checkpoint, the shape in current model is torch.Size([384, 1280]). size mismatch for decoder.rnn_input.weight: copying a param with shape torch.Size([1024, 640]) from checkpoint, the shape in current model is torch.Size([1024, 1152]). size mismatch for decoder.stop_proj.weight: copying a param with shape torch.Size([1, 1536]) from checkpoint, the shape in current model is torch.Size([1, 2048]).