Now I only load part of my train set. I want to get the same part of my train set at every epoch. Is there any chance that I can set epoch concept.
My code like this:
I used this to get my dataloader
return DataLoader(
        dataset,
        batch_size=args.batch_size,
        shuffle=args.shuffle,
        num_workers=args.workers,
        pin_memory=True
    )

use this to get part of train set:
 for (input_tensor, target) in islice(train_loader,self.data_size):
    #my train code