From 48dc6ca157c916156ff4d7e66e699fb8ffb64888 Mon Sep 17 00:00:00 2001 From: Nicolas Kuske Date: Thu, 19 Sep 2024 11:52:56 +0200 Subject: [PATCH] corrected order of valid and train data arguments in DomainDataModule (#82) * swap order of valid and train arguments for class DomainDataModule * also change for GWDataModule --------- Co-authored-by: bdvllrs --- docs/shimmer_basics.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/shimmer_basics.md b/docs/shimmer_basics.md index ce1a4788..40b85bac 100644 --- a/docs/shimmer_basics.md +++ b/docs/shimmer_basics.md @@ -65,8 +65,8 @@ from torch.utils.data import DataLoader, TensorDataset class DomainDataModule(LightningDataModule): def __init__( self, - val_dataset: torch.Tensor, train_dataset: torch.Tensor, + val_dataset: torch.Tensor, batch_size: int, ) -> None: super().__init__() @@ -328,8 +328,8 @@ from shimmer import RepeatedDataset class GWDataModule(LightningDataModule): def __init__( self, - val_datasets: dict[frozenset[str], DomainDataset], train_datasets: dict[frozenset[str], DomainDataset], + val_datasets: dict[frozenset[str], DomainDataset], batch_size: int, ) -> None: super().__init__()