Merge branch 'dev' of https://github.com/ctr26/bioimage_embed into dev

uhlmanngroup · Aug 26, 2023 · 85b7b9d · 85b7b9d
2 parents 4fc2fba + cd443ae
commit 85b7b9d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/bio_vae/lightning/dataloader.py b/bio_vae/lightning/dataloader.py
@@ -67,6 +67,11 @@ def splitting(self, dataset, split_train=0.8, split_val=0.1, seed=42):
  def setup(self, stage=None):
  self.train, self.val, self.test = self.splitting(self.dataset)
 
+ # self.test = self.get_dataloader(test)
+ # self.predict = self.get_dataloader(predict)
+ # self.train = self.get_dataloader(train)
+ # self.val = self.get_dataloader(val)
+
  def test_dataloader(self):
  return DataLoader(self.test, **self.data_loader_settings)