size mismatch for embedding_layer.weight: copying a param with shape torch.Size([X, X])，加载模型出错！！！_size mismatch for vit.patch_embedding.position_emb

作者：小蓝xlanll | 2024-04-28 01:39:12

踩

size mismatch for vit.patch_embedding.position_embeddings: copying a param w

Pytorch,报错size mismatch for embedding_layer.weight: copying a param with shape torch.Size([X, X])，加载模型出错！！！

用LSTM分类时出现错误
在这里插入图片描述
词嵌入层69533与79115不匹配

最后发现是词嵌入字典个数出现错误
当前加载的是79115个单词，而load的模型是经过69533个单词训练的模型，修改成之前相同的数据，字典树就是79115和原来一样了，最后模型加载成功！！！
在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小蓝xlanll/article/detail/499774