这个屌丝很懒，什么也没留下！

RuntimeError: CUDA error: device-side assert triggered排错_embedding runtimeerror: cuda error: device-side as

作者：小蓝xlanll | 2024-02-16 00:56:15

embedding runtimeerror: cuda error: device-side assert triggered

这个问题我的产生是由于embedding层数不一致造成的，

self.word_embeddings = nn.Embedding(config.vocab_size,config.hidden_size,padding_idx=config.pad_token_id)
1

定义的self.word_embeddings的对应值为

(word_embeddings): Embedding(6930, 768, padding_idx=0)
1

而实际上vocab.txt的词汇表有9448个单词

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