RuntimeError: cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/THCCac
Pytorch错误定位:
跑代码时出现了以上错误,发现embedding的向量字典个数设置小于实际的类别数。
另外,使用GPU时对于代码错误的定位是有误的,需要在cpu模式下找到实际的错误定位。
首先看下Embedding层的参数,num_embeddings要设置成字典的大小,但是要记住,一定是vocab_size + 1,而不是vocab_size,即要设置为字典大小+1。参考:http...
原创
2019-09-15 00:24:27 ·
2575 阅读 ·
2 评论