From bf4b668a6cf76582018d25fe4f507ba41990bb64 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Fri, 7 Jun 2024 20:21:21 +0200
Subject: [PATCH] [models.model] Explicitly set the datamodule batch-size to 1
to avoid issues when running in an combo-experiment
---
src/mednet/scripts/saliency/generate.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/mednet/scripts/saliency/generate.py b/src/mednet/scripts/saliency/generate.py
index 71f248fa..649ad96b 100644
--- a/src/mednet/scripts/saliency/generate.py
+++ b/src/mednet/scripts/saliency/generate.py
@@ -180,6 +180,7 @@ def generate(
datamodule.cache_samples = cache_samples
datamodule.parallel = parallel
datamodule.model_transforms = model.model_transforms
+ datamodule.batch_size = 1
datamodule.prepare_data()
datamodule.setup(stage="predict")
--
GitLab