Issue when Re-implement Matrix Factorization in Pytorch

Question

I try to implement matrix factorization in Pytorch as the data extractor and model.

The original model is written in mxnet. Here I try to use the same id