当前位置:   article > 正文

错误处理--CUDA error: device-side assert triggered(很有效)_embeddings _ api . py [ line :39]- error : cuda er

embeddings _ api . py [ line :39]- error : cuda error : device - side as

embeding中词的总数设小了

在函数 nn.Embedding(vocab_size, embedding_dimension) 中

参数vocab_size 为你字典的总词数,如果在训练中有编号>len(字典)的数字出现,就会报这个错表示越界,这通常是因为先前设置nn.Embedding 的最大词数后,后续又加入新词导致的。怎么样是不是被我猜对啦?哈哈因为我也被它折磨了一下午。现在恭喜你已经解决这个问题啦可以继续开心炼丹啦。祝你学习快乐哟~

最后蟹蟹这个答主解开了我的疑惑

CUDA error: device-side assert triggered解决方案 - 知乎

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/89115
推荐阅读
相关标签
  

闽ICP备14008679号