def __getitem__(self, index):
    return {DefaultDataKeys.INPUT: torch.rand(3, *self.size), DefaultDataKeys.TARGET: torch.randint((self.num_classes - 1), self.size)}
def __getitem__(self, index):
    return {DefaultDataKeys.INPUT: torch.rand(3, *self.size), DefaultDataKeys.TARGET: torch.randint((self.num_classes - 1), self.size)}