I have implemented it for my problem and I got this training history (the x axis goes for each epoch through each batch and plots the loss on the y axis) Now I will make some grid for the hyper parameters. Again thanks for all the help.