As criterion I use nn.NLLLoss.
How can I fix the worng number of classes?
I should have 225 classes so I’m not sure why its already failing at 10 classes.
Thank you for that hint, I changed the forward line