Could you tell me the hardware information for the experiment? about knowledge_graph_based_intent_network HOT 4 CLOSED

@huangtinglin

from knowledge_graph_based_intent_network.

Sorry for the late reply. The experiments are conducted on a single Linux server with 512G RAM and 8 NVIDIA GeForce v100-32GB.

from knowledge_graph_based_intent_network.

Thank you for your reply.In the training log of Amazon-Book dataset, the training time of eachepoch is 800+ seconds, which is unacceptable. I used to use the whole last.FM Data sets (3 million interactions and 460000 + triplets) and also use Tesla V100 to train KGAT, and the average train time of each epoch is only 182 seconds. KGIN has no additional knowledge Graph representation learning(e.g. transR). Why does it spend so much time training?

from knowledge_graph_based_intent_network.

I think the most time-consuming procedure is calculating the equation (7)(10), which is implemented in the "KGIN" file (line 35-38). Here I use the torch_scatter package to aggregate the neighbor embedding, and it can be further optimized or replaced by another efficient implementation.

from knowledge_graph_based_intent_network.