Tests: refactor cleanup (#1744)

* wip

* cleaning

* optim imports

* -

* default hparams

* fix restore

* fix imports
This commit is contained in:
Jirka Borovec
2020-05-10 13:15:28 -04:00
committed by GitHub
parent 4970927ec8
commit 134eb61e1a
41 changed files with 187 additions and 1150 deletions
+20
View File
@@ -0,0 +1,20 @@
from abc import ABC, abstractmethod
from tests.base.dataloaders import CustomInfDataloader
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)]
def val_dataloader__infinite(self):
return CustomInfDataloader(self.dataloader(train=False))