Thanks for the information!


It seems dim1 is unnecessary, but based on the current shape, it doesn’t matter if you squeeze dim0 or dim1.


Yes, I would just like to know, if the model performance is already bad during training and if the model also outputs a single class then.


It looks like you are also normalizing, which would make a difference, if only applied during testing.