generally, we use the entire (partly) trained model to run on the val sets.
if the input shape of the train and val set have difference, u can change it to be consistent.
this is my suggestion.
hope that others give a good solution for this problem.