neural_graph_collaborative_filtering
neural_graph_collaborative_filtering copied to clipboard
how to test the performence in the test dataset
why the training process so slow
In my experience, u need a GPU. Tested on 2080ti, 20s for 1 epoch while without GPU, it took 800s 1 epoch. And by the way, 99% the training time is used in sample process.
why did i often occur "loss is nan"
You can try the following code to modify the BPR loss, in order to avoid NAN:
loss = tf.reduce_sum(tf.nn.softplus(-(pos_result - neg_result)))
you mean tf.reduce_mean?
why did i can only use 149M GPU memory, which has a full memory of 10GB
I have no idea... Have you installed the tensorflow-gpu successfully?
I have no idea... Have you installed the tensorflow-gpu successfully? i fix it . the reason is that i install tensorflow-gpu 1.14.0,when i change it to tensorflow-gpu 1.8.0, it works