from torch.utils.data import DataLoader class PrivacyDataloader(DataLoader): def __init__(self, *args, **kwargs): super(PrivacyDataloader, self).__init__(*args, **kwargs)
