From cf74231796b4885e3168b319838e46c3f8f2efcd Mon Sep 17 00:00:00 2001
From: Gokhan Ozbulak <gokhan.ozbulak@idiap.ch>
Date: Fri, 24 May 2024 11:50:14 +0200
Subject: [PATCH] Fix typo on calling sub-field. #25
---
src/mednet/data/datamodule.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/mednet/data/datamodule.py b/src/mednet/data/datamodule.py
index d7b6e93b..56ab3000 100644
--- a/src/mednet/data/datamodule.py
+++ b/src/mednet/data/datamodule.py
@@ -792,7 +792,7 @@ class ConcatDataModule(lightning.LightningDataModule):
return torch.utils.data.DataLoader(
self._datasets["train"],
shuffle=(self._train_sampler is None),
- batch_size=self.batch,
+ batch_size=self.batch_size,
drop_last=self.drop_incomplete_batch,
pin_memory=self.pin_memory,
sampler=self._train_sampler,
--
GitLab