After a few epochs, the model does not learn any features and displays same output after every epoch. There must be an error in the code.