Variational Generator

2020-03-10 16:59:51 +01:00
parent 21e7e31805
commit 1b5a7dc69e
10 changed files with 177 additions and 95 deletions
@@ -154,7 +154,7 @@ class LightningBaseModule(pl.LightningModule, ABC):

    # Validation Dataloader
    def val_dataloader(self):
-        return DataLoader(dataset=self.dataset.val_dataset, shuffle=False,
+        return DataLoader(dataset=self.dataset.val_dataset, shuffle=True,
                          batch_size=self.hparams.train_param.batch_size,
                          num_workers=self.hparams.data_param.worker)