search

hidasib / GRU4Rec

GRU4Rec is the original Theano implementation of the algorithm in "Session-based Recommendations with Recurrent Neural Networks" paper, published at ICLR 2016 and its follow-up "Recurrent Neural Networks with Top-k Gains for Session-based Recommendations". The code is optimized for execution on the GPU.
Other
747 stars 222 forks source link

About training time #28

Closed DebonairLi closed 4 years ago

DebonairLi commented 5 years ago

Sorry to disturb. I've used Titan Xp to run your code, but it needs four hours for each epoch, that is for the final results, I need to wait about two days or more. Du you know why? thanks.

hidasib commented 4 years ago

It should have been much faster, maybe something in the config was off. Check out the recent update, it is even faster and it is also easier to use and set up. Also, some info about training times is now included in the readme.