cripac-dig / lattice Goto Github PK

您好，非常感谢能够看到这样优秀的工作，有一点想要请教，我把项目代码放在3090显存为24G的单卡上运行，出现cuda out of memory的错误，故想向作者询问一下在训练时使用的显卡，以及代码需要多大的显存才能运行。

Hi, thanks for your amazing work.

Just want to ask the baby dataset result is showed as from the code :
{'precision': array([0.00545127, 0.00432502]), 'recall': array([0.05198949, 0.08228838]), 'ndcg': array([0.02913475, 0.03712578]), 'hit_ratio': array([0.05425559, 0.08593469]), 'auc': 0.0}

but it is quite different from what you reported in the paper, just wonder if this is the final code of your implementation?