lium-lst / nmtpytorch

Sequence-to-Sequence Framework in PyTorch
Other
392 stars 51 forks source link

Working on images #11

Closed bertanunez closed 6 years ago

bertanunez commented 6 years ago

Hello Ozan, until today I have been working only with the translation of the sentences and everything works fine. But when I try to work with images the problems appear. I've seen you ipynb for attention visualization but when running the process function i get these error referred to the batch (imgs = batch['image'].data.cpu())

KeyError: 'image'

The batch I get is the next one and the only indices are 'en' and 'fr'.

When creating the model I doesn't mention anything about images, so what are the steps that I am skipping?