The evaluation when runnin test.py
Timon0327 opened this issue · 3 comments
Timon0327 commented
Hi. When I run the test.py to evaluate, the model just generates <'unk'> without any other tokens.
jianqingxie commented
me too, do you solve it?
jianqingxie commented
Timon0327 commented
Hi, because I had a bug about gather_cuda() from beam_search.py, I modified the data.dtype to float.64 to solve it. When I changed it to original version, the test was normal.