PyTorch: Using a target size (torch.Size([1])) that is different to the input size (torch.Size([1, 1]))

Question

I am new to PyTorch and working on the implementation of recommender systems.

I obtained my models from here: https://blog.fastforwardlabs.com/2018/04/10/pytorch-for-reco