Closed HUANGYUEXIN closed 4 years ago
1、能否提供一下确切的错误信息,数据文件的目录组织请查看 #1 2、是否是模型有误?请您查看一下bird的text_encoder200.pth,encoder.weight的shape是否为[5450, 300]。text_encoder200.pth的md5值应当为11fe9392efb4f231fd29f2cf77ff357e。
在复现过程中,我出现了两个问题: 1、严格按照说明放置模型文件和数据文件,出现了assert datasets错误。 2、出现了维度不匹配问题,不知道是模型位置问题还是模型提供错误。 RuntimeError: Error(s) in loading state_dict for RNN_ENCODER: size mismatch for encoder.weight: copying a param of torch.Size([1, 300]) from checkpoint, where the shape is torch.Size([5450, 300]) in current model.“data” 我也出现了和你一模一样的问题,请问是怎么解决的呢?作者可以回复一下吗?
Yes, I got the same problem "RuntimeError: Error(s) in loading state_dict for RNN_ENCODER:size mismatch for encoder.weight: copying a param of torch.Size([1, 300]) from checkpoint, where the shape is torch.Size([5450, 300]) in current model.". So how could I fix it?
在复现过程中,我出现了两个问题: 1、严格按照说明放置模型文件和数据文件,出现了assert datasets错误。 2、出现了维度不匹配问题,不知道是模型位置问题还是模型提供错误。 RuntimeError: Error(s) in loading state_dict for RNN_ENCODER: size mismatch for encoder.weight: copying a param of torch.Size([1, 300]) from checkpoint, where the shape is torch.Size([5450, 300]) in current model.“data”