@smth yes, you are right. I did try with lr=0.0001 and the training loss didn’t explode much in one of the epochs. however this second experiment I did increase the number of filters in the network.

training_loss_Exp2.png700×450 22.8 KB