Files
pytorch-lightning/tests/base/eval_model_valid_dataloaders.py
T
Jirka Borovec 6d58fb1353 Tests: refactor trainer (#1728)
* lr

* optim

* wip

* wip

* fix mean

* flake8
2020-05-04 16:51:39 -04:00

16 lines
368 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)
def val_dataloader__multiple(self):
return [self.dataloader(train=False),
self.dataloader(train=False)]