I am new to PyTorch and trying to implement a recommender system. I am working on my first model (matrix factorization) which i found here: https://blog.fastforwardlabs.com/2018