首先是gpu报错
其实这个只是gpu错误显示不全,所以显示出来这个
然后换cpu跑,就会显示出来详细错误经过仔细查找,发现是embedding层src_vocab_size指的输入序列中不同单词的个数,我忘记算上padding了,所以我要+1,就成功运行啦
首先是gpu报错
其实这个只是gpu错误显示不全,所以显示出来这个
然后换cpu跑,就会显示出来详细错误经过仔细查找,发现是embedding层src_vocab_size指的输入序列中不同单词的个数,我忘记算上padding了,所以我要+1,就成功运行啦