Would it be possible to upload the test_loader so that I could run your code?
If it’s too big, maybe a small part would be sufficient (e.g. 1000 samples).
Currently I can’t locate the source of this issue.