@Dwight_Foster  Thanks…That was due to shape of data returned from data loader
Actually the shape of data from dataloader was not correct
Thank you so much!!
Also…I hope my model layers are proper in place …I m pretty new to this stuff