Multi Layer Perceptron Deep Learning in Python using Pytorch

Question

I am having errors in executing the train function of my code in MLP.

This is the error:

mat1 and mat2 shapes cannot be multiplied (128x10 and 48x10)

My