devforfu/pytorch_playground

A problem in training

Opened this issue · 1 comments

Hi
in forward function you first get users and then movies:
def forward(self, users, movies, minmax=None):
but in training you first get column 1 which is movies id and then column 0 which is user id:
outputs = net(x_batch[:, 1], x_batch[:, 0], minmax)
I wanted to know if there is a reason behind this or it is a problem.
thank you very much for your time

outputs = net(x_batch[:, 1], x_batch[:, 0], minmax)
Just switch between 0 & 1. And everything will be fine.