babysor / babysor/MockingBird

RuntimeError: Error(s) in loading state_dict for Tacotron:

Open
#903 5 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
36.9k
Forks
5.2k
PR merge metrics
No merged PRs in 30d

Description

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]).
![image](https://github.com/babysor/MockingBird/assets/109200167/494cf902-6efb-4416-995d-b97d2935d70b)

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reproducing the checkpoint-loading failure and compare the saved Tacotron parameter shapes with the current model configuration. Trace the encoder and decoder definitions to identify why the checkpoint and model dimensions differ; done means the reported checkpoint loads without size-mismatch errors.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.