As it is, it is difficult to reproduce this output because we don’t know the directory structure that you are working with and there is a lot of code unrelated to the cross-validation/dataloading process. Can you give an example of the directory structure and a minimal example that presents the issue?