Sorry, maybe I was not clear. I have this dataset (Let’s call it A) split into train and validation. When I train the two sets of A dataset (see train function above) everything works correctly but when I want to test the model on a different image dataset it returns different results every time. For example for one image I might get labels like clouds, storm etc. and if I run the model again I get different labels for the same image.