anil_batra:

I suggest you to please lower the batch size (probably half it) to train the code


That means that I should change the batch size to 8 (initial 16) and start training from the beginning right? Is it possible to continue training with the best so far model, using a different batch size?