Of course,
So I’m using WeightedRandomSampler with replacement = False and setting the weights for each image in the dataset to equal 1. This way the DataLoader will select all the images over one epoch but given that I can restrict the num_samples, I can observe how the CNN performs on a test set before it’s seen all the data.
This was the only way that I could restrict the number of samples because I couldn’t find a num_samples option in DataLoader.
Does this make sense?