maciejkula / spotlight

Deep recommender models using PyTorch.
MIT License
2.97k stars 421 forks source link

Refactoring the code #163

Open hadi-gharibi opened 5 years ago

hadi-gharibi commented 5 years ago

Hello, I was wondering if you have any plan to refactor the code and make it more look like a framework like fastai, ignite, etc. Right now we have to implement the training loop for each new model and this is not efficient. If we could use callbacks, it would be more easy to use more complicated training approaches(like anneal learning rate) and at the same time use something like TensorFX and even logging without headache.