You are not computing the loss on the test set

[CAridorc]

If I understood your code correctly, are not computing the loss on the test set, so you cannot know if you have overfitting to the training data.