Solved by Oli in post #2 
              
              
                Hello, I would advise you to create one dataloader for training and one for testing. To do this you can implement your own dataset which reads training data and corresponding label. There are a bunch of examples of such datasets online but here is one from the tutorials 
Your dataset would know abou…