Thanks for the response,
I did the changes, but it doesn’t seem to be working. even if the network is not efficient, i expect it should give some decent prediction. I can see only 50% accuracy which no better than random guess.
Could there be some problem with my validation part of the code ?