babysor / babysor/MockingBird

模型加载错误,维度错误

Open
#899 6 comments 1 reaction 0 assignees View on GitHub
Dominant language
Python
Stars
36.9k
Forks
5.2k
PR merge metrics
No merged PRs in 30d

Description

使用的最新的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]).

Contributor guide

No contributing guide indexed for this repository

Research direction

Start with the Tacotron state_dict loading entry point indicated by the error, then compare the model and checkpoint dimensions against the Quick Start (Newbie) wiki and issue #37. Done means the documented synthetic model loads without the reported embedding and decoder size mismatches.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
ai, machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.