Training Always Stops After 3 Epochs

[Supersak80]

Hello,

I'm running the code to train on the various toy datasets (e.g., Cifar10, Mnist, etc.) and every time I train, regardless of the # of epochs I specify, the code runs for only 3 epochs and halts. Any idea why?