search

MILVLG / bottom-up-attention.pytorch

A PyTorch reimplementation of bottom-up-attention models
Apache License 2.0
294 stars 76 forks source link

Library versions #71

Closed realjoenguyen closed 3 years ago

realjoenguyen commented 3 years ago

Dear all,

I had some trouble installing the decaltron2 in the code and I've just figured it out it was due to the library version mismatch between libraries the author used and my enviroment. I post pytorch-related library versions that works for the code here in case you had the same problem:

For CUDA 10.1: torch==1.6.0+cu101 torchvision==0.7.0+cu101

Thanks,

Zoroaster97 commented 3 years ago

👍