Hey,

Thanks for reaching out.

Where are you loading your data from?

Are the only differences between the two trials the size of the dataset? Everything else is the same (data location, file sizes, file format, pre-processing/augmentation pipeline etc.).

Is it possible that the small dataset is beng cached in memory?