remove redundant dataloader init

teamtomo · Sep 21, 2024 · 0ac5a41 · 0ac5a41
1 parent a4a9ff0
commit 0ac5a41
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/membrain_seg/segmentation/train.py b/src/membrain_seg/segmentation/train.py
@@ -95,7 +95,6 @@ def train(
         on_the_fly_dataloading=on_the_fly_dataloading,
         aug_prob_to_one=aug_prob_to_one,
     )
-    data_module.setup(stage="fit")
 
     # Set up the model
     model = SemanticSegmentationUnet(