From d82428376aabc977eb75928cc47bd5dc55b7b129 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Wed, 5 Jun 2024 12:04:39 +0200
Subject: [PATCH] [classification.scripts] Fix call to setup_datamodule
---
src/mednet/libs/classification/scripts/train.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/mednet/libs/classification/scripts/train.py b/src/mednet/libs/classification/scripts/train.py
index d1a13769..8483f667 100644
--- a/src/mednet/libs/classification/scripts/train.py
+++ b/src/mednet/libs/classification/scripts/train.py
@@ -75,6 +75,7 @@ def train(
setup_datamodule(
datamodule,
model,
+ batch_size,
drop_incomplete_batch,
cache_samples,
parallel,
--
GitLab