Heeello,
I agree with @tymokvo that more background into the problem helps (or an explanation on why this isn’t needed).
Luckily I found one possible error in your code. Your forward/classifier function squeezes the output into a softmax whilst your loss function,  cross_entropy, doesn’t want this. See the docs about cross entropy and NLLLoss 206