def val_dataloader__infinite(self):
     return CustomInfDataloader(self.dataloader(train=False))
 def train_dataloader__infinite(self):
     return CustomInfDataloader(self.dataloader(train=True))