Solved by Henry_Chibueze in post #27 
              
              
                Yo, so I’ve fixed the problem finally 
The issue was that the NNModel.train() was outside the epoch loop, and in the epoch loop, the testing function is called at the 0th epoch and it toggles the NNModel.train() to NNModel.eval(). 
So since the NNModel.train() was outside the loop, after the 0th epo…