2020-04-16 02:16:40 +00:00
|
|
|
from abc import ABC, abstractmethod
|
|
|
|
|
|
2020-05-10 17:15:28 +00:00
|
|
|
from tests.base.dataloaders import CustomInfDataloader
|
2020-05-05 16:31:15 +00:00
|
|
|
|
2020-04-16 02:16:40 +00:00
|
|
|
|
|
|
|
|
class ValDataloaderVariations(ABC):
|
|
|
|
|
|
|
|
|
|
@abstractmethod
|
|
|
|
|
def dataloader(self, train: bool):
|
|
|
|
|
"""placeholder"""
|
|
|
|
|
|
|
|
|
|
def val_dataloader(self):
|
|
|
|
|
return self.dataloader(train=False)
|
2020-05-04 20:51:39 +00:00
|
|
|
|
|
|
|
|
def val_dataloader__multiple(self):
|
|
|
|
|
return [self.dataloader(train=False),
|
|
|
|
|
self.dataloader(train=False)]
|
2020-05-05 16:31:15 +00:00
|
|
|
|
|
|
|
|
def val_dataloader__infinite(self):
|
|
|
|
|
return CustomInfDataloader(self.dataloader(train=False))
|
