search

facebookresearch / deepcluster

Deep Clustering for Unsupervised Learning of Visual Features
Other
1.69k stars 325 forks source link

Problems when using faiss in this code #57

Closed accwht closed 4 years ago

accwht commented 4 years ago

Hi, I am interested in this project and trying to use the model to train my own image data. However, I found that the faiss tool with GPU is very slow in the starting time. The code "return _swigfaiss.PCAMatrix_train(self, n, x)" costs over nearly 20 minutes. Therefore I wonder if there is anyway to solve this problem and speed up the training process? Thanks a lot!