RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/THCCac_cuda runyime error (59)

作者：笔触狂放9 | 2024-02-16 00:55:07

踩

cuda runyime error (59)

Pytorch错误定位：

跑代码时出现了以上错误，发现embedding的向量字典个数设置小于实际的类别数。

另外，使用GPU时对于代码错误的定位是有误的，需要在cpu模式下找到实际的错误定位。

首先看下Embedding层的参数，num_embeddings要设置成字典的大小，但是要记住，一定是vocab_size + 1，而不是vocab_size，即要设置为字典大小+1。参考：https://blog.csdn.net/Geek_of_CSDN/article/details/86527107

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