inference error Assign requires shapes of both tensors to match. lhs shape= [5472,410] rhs shape= [84827,410] about transformer-xl-chinese HOT 7 OPEN

训练doupo没有改任何参数，推断出现同样的问题

from transformer-xl-chinese.

我在训练doupo的时候,也是出现了同样的问题.

from transformer-xl-chinese.

楼上的两位，请问问题解决了没

from transformer-xl-chinese.

我遇到了类似的问题发现是因为词表不一致的问题，可以看一下 train_gpu.py 文件中，inference函数的dataset_name与train函数是否一致？

from transformer-xl-chinese.

我遇到了类似的问题发现是因为词表不一致的问题，可以看一下 train_gpu.py 文件中，inference函数的dataset_name与train函数是否一致？

什么意思？与train函数的什么是否一致？

from transformer-xl-chinese.

抱歉各位，回复的比较晚，大家看下readme 中”注意在inference的时候记得修改train_gpu.py中第504行，改成你想inference的数据集名字“ 有没有注意到

from transformer-xl-chinese.

各位 不知道你们还在不在这个repo上跟进，回复各位较晚了 非常抱歉，问题已经修复，感谢你们的反馈， 问题来源于之前在data/doupo/ 下有一个记录corpus 的信息的json，和一个cache.pkl， 这样在运行bash base_doupo_gpu.sh train_data 时,如果识别到cache.pkl， 就不会重新生成训练数据，会直接使用原来的有问题的那个，所以简单的解决办法就是把除*.txt 以外的都直接删掉即可。

from transformer-xl-chinese.