neuroailab / LocalAggregation-Pytorch

88 stars 17 forks source link

AttributeError: 'Clustering' object has no attribute 'obj' #4

Open pawandeep2155 opened 4 years ago

pawandeep2155 commented 4 years ago

When I run command CUDA_VISIBLE_DEVICES=0 python scripts/localagg.py ./config/imagenet_la.json to do LA training, I get the following error :

LocalAggregation-Pytorch/src/objectives/localagg.py(240)run_kmeans() -> losses = faiss.vector_to_array(clus.obj) (Pdb) n AttributeError: 'Clustering' object has no attribute 'obj'

I checked the documentation of Faiss and there seems to be no attribute obj. Can you please check it?

chengxuz commented 4 years ago

I believe a recent update of faiss changed its interface. You need to install faiss_gpu==1.6.1, see this issue: https://github.com/neuroailab/VIE/issues/7. I will add this to the documentation, thanks!

pawandeep2155 commented 4 years ago

Thanks for your reply. I installed faiss version 1.6.1 and it worked.