错误处理--CUDA error: device-side assert triggered(很有效)
点进来的朋友你好啊,先别说,让我猜猜,你是不是正在写一个NLP模型?它是不是有lstm,或者transformer,或者textcnn?哈哈是不是猜中啦。你现在一定很着急,看了各种博客,说是下标越界~,可是找来找去,真的找不出哪里出界了?打印出来各个都在标签限制内呢?别着急,我相信看了我的博客有很大概率可以解决你的难题。那就是————————————————————你的embeding中词的总数设小啦!在函数 nn.Embedding(vocab_size, embedding_dim
原创
2022-04-09 00:23:14 ·
4420 阅读 ·
1 评论