huangjunjie-cs / LDA-GCL

source code for LDA-GCL(DASFAA 2023)
https://arxiv.org/abs/2302.02317
6 stars 0 forks source link

关于代码的一些疑问 #3

Open TJWHJ opened 8 months ago

TJWHJ commented 8 months ago

作者你好,我有幸拜读了你们的这篇文章,觉得很精彩。我尝试复现你们的实验,目前有两个疑问:

  1. 预训练的lightGCN模型和LDA-GCL使用的是同一划分(data splitting)下的数据集吗?
  2. 我使用你们保存的lightGCN模型,和提供的划分好的gowalla数据集,进行复现(所用显卡是NVIDIA Tesla V100),发现平均一个epoch的训练时间要590s左右,请问这是正常的时间消耗吗? 期待作者的解答,感谢!
huangjunjie-cs commented 8 months ago

感谢你对我们项目的兴趣。

  1. 是的
  2. 可能是的,不过对比需要似乎20左右epoch就收敛了,具体时间我忘记了