pmixer / TiSASRec.pytorch

https://github.com/JiachengLi1995/TiSASRec in PyTorch
84 stars 15 forks source link

Running time #5

Closed xiqxin1 closed 2 years ago

xiqxin1 commented 2 years ago

Hi, Thank you sooo much for this code. Did you remember how long it takes for one epoch? I'm not sure if there's bug of my implementation.

pmixer commented 2 years ago

longer than SASRec.pytorch, but should not take too long, maybe try debugging with help from import pdb; pdb.set_trace()

xiqxin1 commented 2 years ago

Thanks a lot! I'm now debugging it.