Model shape mismatch

[firstprayer]

Summary[问题简述（一句话）]
用Pretrain模型(https://yisiou-my.sharepoint.com/:u:/g/personal/lawrence_cheng_fawenyo_onmicrosoft_com/EWFWDHzee-NNg9TWdKckCc4BC7bK2j9cCbOWn0-_tK0nOg?e=n0gGgC)，加载模型的时候出现shape mismatch

Env & To Reproduce[复现与环境]
模型链接如上
在Mac M1 Pro，环境Python3.9

Screenshots[截图（如有）]

RuntimeError: Error(s) in loading state_dict for Tacotron:
	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]).

这已经是按照#37 修复过一轮了，否则还要再多一个类似的error。

[firstprayer]

Resolved after using the latest retrained model