search

yufengm / Adaptive

Pytorch Implementation of Knowing When to Look: Adaptive Attention via A Visual Sentinel for Image Captioning
107 stars 42 forks source link

train.py #12

Open gorgeousyouth opened 5 years ago

gorgeousyouth commented 5 years ago

Hi, this is the best project, but when i use the train.py the error occurs. Here is the error RuntimeError: matrix and matrix expected at /opt/conda/conda-bld/pytorch_1501971235237/work/pytorch-0.1.12/torch/lib/THC/generic/THCTensorMathBlas.cu:237 My pytorch version is 0.1.12 Best,