this is overall model shape so I am intially passing images of [3, 250, 250] with 2 classes and once tge model is trained I want to retrain it using smaller amount of dataset but it is giving me error…