Thanks for your response.
Response to


I checked the model and the unique number of labels in the prediction during each training iteration.
it seems like the model has both labels 0 and 1 during the initial stages of training, once the model reaches 99% accuracy further training for some reason changes the number of unique labels in prediction to only 1 which is label 0. Also, the 99% accuracy is obtained half way though the first epoch. I am only training for 1 epoch


I changed my code to only not have normalization for training and testing.


Thanks for your help
Nishanth