issue on testing lgcn about learn-to-cluster

[yl-1993]

Hi @eezxuan , sorry for the late reply and thanks for reporting the issue.

Would you mind trying the provided faiss knns first? One potential problem is the inconsistent knn construction method between training and test. We have observed the performance drop when we use hnsw for training and use faiss for testing.

But we have not do not try your setting yet, we may try it first and report here later.

@eezxuan I have one small question. Which version of Pytorch and Cuda are you using to run the particular pretrained model. I am constantly having issues in Google Colab. Thanks in advance !

[yl-1993]

@redionxhepa Hi, for you knowledge, I used torch==0.5.0a0+e31ab99 and cuda 9.0 for experiments. Feel free to report issues encountered for different package versions.

I tried a part of the code and it works even in the new Colab Environments which I think have 1.x Pytorch versions and way more newer cuda versions. Thank you and nice job !