Lightning acc

Closes #25 (closed)

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,