Last_One:

I always have errors on the network, in particular with image or label shapes


If you have size mismatches during forward pass you might want to look at your featuremaps. Check whether or not the out_channels and in_channels of your conv layers match as well as the in_features and out_features of your linear layer. Just printing x.shape() at some points might help you find and debug mismatches