mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-13 12:50:26 +08:00
Tests: refactor cleanup (#1744)
* wip * cleaning * optim imports * - * default hparams * fix restore * fix imports
This commit is contained in:
@@ -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))
|
||||
Reference in New Issue
Block a user