sorry for the late reply, i was a bit caught up, I have fixed this issue the problem turned out to the resume training part of code,which was linking to a different model path , and thats why i was getting sizes dont match.
But anyways thank you for your time.