Yes, actually I am doing pre-training for the model and I wish to use complete dataset during this part.
But I am not sure how to achieve this using ConcatDataset.
I am using MNIST dataset and I have loaded train_loader and test_loader using torchvision