Add keepdim=True for PyTorch version upgrade to 0.2.0

Cyanogenoid / pytorch-vqa

Strong baseline for visual question answering

238 stars 97 forks source link

Closed pplantinga closed 7 years ago

pplantinga commented 7 years ago

As per issue #1 update code to be compatible with PyTorch 0.2.0

Cyanogenoid commented 7 years ago

Thanks for the contribution!

pplantinga commented 7 years ago

No problem :)