The initial (irrelevant) score is varied and below 40000 (/80000). After further epochs, the result is one of 60303 after every epoch.
Also, I printed the y_pred being emitted for each data point and it always has the value [0.4404, -0.5643] which means it’s always predicting 0.
I’ve tried using NLLLoss, and a few others as well. Same issue.