How is the efficiency of your code?

xiangwang1223 / neural_graph_collaborative_filtering

Neural Graph Collaborative Filtering, SIGIR2019

MIT License

781 stars 261 forks source link

How is the efficiency of your code? #8

Open yuanyuansiyuan opened 5 years ago

yuanyuansiyuan commented 5 years ago

when i run your code on the Gowalla dataset, it seems that it is stucked after printing "without using pretraining". I have configured the GPU in code by: os.environ["CUDA_VISIBLE_DEVICES"] = '0'

but it seems like not using GPU at all. What about your running experience and the code efficienty? Thank you!

xiangwang1223 commented 5 years ago

Hi, I performed experiments on GPU. Please check whether you have installed tensorflow-gpu environment first, and then guarantee such environment activated. Thanks.