mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-12 13:00:12 +08:00
Jirka BorovecandWilliam Falcon
2a2f303ae9
* refactor default model * drop redundant seeds * refactor dataloaders tests * fix multiple * fix conf * flake8 * Apply suggestions from code review Co-authored-by: William Falcon <waf2107@columbia.edu> Co-authored-by: William Falcon <waf2107@columbia.edu>
23 lines
657 B
Python
23 lines
657 B
Python
from abc import ABC, abstractmethod
|
|
|
|
from tests.base.eval_model_utils import CustomInfDataloader
|
|
|
|
|
|
class TrainDataloaderVariations(ABC):
|
|
|
|
@abstractmethod
|
|
def dataloader(self, train: bool):
|
|
"""placeholder"""
|
|
|
|
def train_dataloader(self):
|
|
return self.dataloader(train=True)
|
|
|
|
def train_dataloader__infinite(self):
|
|
return CustomInfDataloader(self.dataloader(train=True))
|
|
|
|
def train_dataloader__zero_length(self):
|
|
dataloader = self.dataloader(train=True)
|
|
dataloader.dataset.data = dataloader.dataset.data[:0]
|
|
dataloader.dataset.targets = dataloader.dataset.targets[:0]
|
|
return dataloader
|