nn.Embedding

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sinat_26114733/article/details/88061889

当数组越界的时候报错是CUDA问题,并且在语句执行的时候不报错而是影响后面的语句执行。

THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1533672544752/work/aten/src/THC/generic/THCTensorCopy.cpp line=20 error=59 : device-side assert triggered

检查embdding的输入是否出现了负数或者超出了词表大小。

猜你喜欢

转载自blog.csdn.net/sinat_26114733/article/details/88061889