KeyError: tensor(2, device=‘cuda:0‘)

在运行pytorch官方的一个教程时出现KeyError: tensor(2, device='cuda:0')错误
解决方法:

将
decoded_words.append(output_lang.index2word[ni])
改为
decoded_words.append(output_lang.index2word[int(ni)

你可能感兴趣的:(python,深度学习,pytorch)