search

sgrvinod / a-PyTorch-Tutorial-to-Image-Captioning

Show, Attend, and Tell | a PyTorch Tutorial to Image Captioning
MIT License
2.76k stars 713 forks source link

Evaluation issue #63

Open anand-kummari opened 5 years ago

anand-kummari commented 5 years ago

Hi, I am new to pytorch. I have trained as you mentioned using flickr8k dataset. After training how to run eval.py? I am running the code in google colab. on running the eval.py i am getting the following error

Traceback (most recent call last): File "/content/gdrive/My Drive/B.Tech Project/Attention/eval.py", line 179, in print("\nBLEU-4 score @ beam size of %d is %.4f." % (beam_size, evaluate(beam_size))) File "/content/gdrive/My Drive/B.Tech Project/Attention/eval.py", line 131, in evaluate incomplete_inds = [ind for ind, next_word in enumerate(next_word_inds) if File "/content/gdrive/My Drive/B.Tech Project/Attention/eval.py", line 132, in next_word != word_map['']] RuntimeError: CUDA error: device-side assert triggered

aliciaviernes commented 2 years ago

Did you manage to solve it? I assume that it has to do with the Wordmap.