Solved by rahulvigneswaran in post #3 
              
              
                Hi @NoobCoder, 
@ptrblck is suggesting something like this, 
class YourCustomSubset(torch.utils.data.Subset):
    r"""
    Same as torch.utils.data.Subset. This outputs the indices of each batch, you can choose which indices to keep.
    """

    def __getitem__(self, idx):
        return self.datas…