加载预训练模型并增加一层nn.Embedding层,运行代码时,抛出RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED的错误。这种错误一般情况下是由于pytorch和CUDA版本不对应引起的,但是预训练模型在同样的环境下能跑通。故,排除版本不匹配问题。经检查,原因是自己增加的nn.Embedding(vocab_size, dim)中的vocab_size=49417设置错误。我的词典从0到49417共有49418个单词,而我一开始设置的是49417,查询表大小错误,将其更改为size应该设置为49418后,错误解决。



