Solved by AladdinPerzon in post #3 
              
              
                The first thing you need to do is to separate your dataset into a training, validation and test set. For each of these you create a loader exactly like you’ve done for the training loader. Assuming you’ve done that and have a training_loader, validation_loader, and test_loader, you could then define…