-
Notifications
You must be signed in to change notification settings - Fork 117
Open
Description
您好,感谢您用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
Labels
No labels