Run time error with GPU

kevinzakka / recurrent-visual-attention

A PyTorch Implementation of "Recurrent Models of Visual Attention"

MIT License

468 stars 123 forks source link

Closed ipod825 closed 6 years ago

ipod825 commented 6 years ago

python main.py --use_gpu 1

RuntimeError: Expected object of type Variable[torch.cuda.FloatTensor] but found type Variable[torch.Flo
atTensor] for argument #1 'mat1'

kevinzakka commented 6 years ago

I'll check it out as soon as I get some free time :)

kevinzakka commented 6 years ago

Okay should be fixed now.

Cheers