Thank you that worked.
I have a couple of questions.
Right now, my model’s accuracy cannot go over 40% after 50 epochs.
Is this an overfitting issue?
I added some dropout layers, but I am not sure how many and what percentage I should be using. Currently I have 8 dropout layers, and 9 batchnorm layers, maybe this is too much?
After some research, I found out that if I use batchnorm, I should set the bias to false, is this the case for every convolution and linear layer?