Skip to content

复现结果 #51

@SwaggyZhang

Description

@SwaggyZhang

您好,感谢您用torch复现了SASRec。

在README.md里面,您提供了直接运行推理的命令:

python main.py --device=cuda --dataset=ml-1m --train_dir=default --state_dict_path='ml-1m_default/SASRec.epoch=1000.lr=0.001.layer=2.head=1.hidden=50.maxlen=200.pth' --inference_only=true --maxlen=200

如果不训练直接推理的话,NDCG和HR指标很低,大概只有标称的一半。

我重新训练后,就可以达到标称数值了,不知您上传的checkpoint是否有误?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions