meshed-memory-transformer icon indicating copy to clipboard operation
meshed-memory-transformer copied to clipboard

Published 20 hours ago • verified publisher icon aimagelab

The evaluation when runnin test.py

Open Timon0327 opened this issue 3 years ago • 3 comments

Hi. When I run the test.py to evaluate, the model just generates <'unk'> without any other tokens.

Timon0327 avatar Dec 24 '21 07:12 Timon0327

me too, do you solve it?

jianqingxie avatar Mar 06 '22 07:03 jianqingxie

@Timon0327

jianqingxie avatar Mar 06 '22 08:03 jianqingxie

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.

Timon0327 avatar May 17 '22 12:05 Timon0327