userEmbeddings = self.userEmbeds[userIndex] movieEmbeddings = self.movieEmbeds[movieIndex]
To -->
userEmbeddings = self.userEmbeds(userIndex) movieEmbeddings = self.movieEmbeds(movieIndex)can't covert CUDA tensor to numpy. use Tensor.cpu() to copy the tensor to host memory first.
Change
index = output.data.numpy().argmax()
to
index = output.cpu().data.numpy().argmax()
convert list to tensor: torch.Tensor(a)
