aimagelab / meshed-memory-transformer

Meshed-Memory Transformer for Image Captioning. CVPR 2020

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The evaluation when runnin test.py

Timon0327 opened this issue · comments

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

me too, do you solve it?

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.