lightning/tests/base/eval_model_valid_dataloader...

12 lines
235 B
Python

from abc import ABC, abstractmethod
class ValDataloaderVariations(ABC):
@abstractmethod
def dataloader(self, train: bool):
"""placeholder"""
def val_dataloader(self):
return self.dataloader(train=False)