From 0f069af7a26ecd965a3a534b2a0266a994d38f2a Mon Sep 17 00:00:00 2001 From: camillebrianceau Date: Mon, 14 Oct 2024 10:24:05 +0200 Subject: [PATCH] remove split list --- clinicadl/trainer/trainer.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/clinicadl/trainer/trainer.py b/clinicadl/trainer/trainer.py index a949e1301..47f31ee5a 100644 --- a/clinicadl/trainer/trainer.py +++ b/clinicadl/trainer/trainer.py @@ -220,7 +220,7 @@ def train( else: splitter_config = SplitterConfig(**self.config.get_dict()) - split_manager = Splitter(splitter_config, split_list=split_list) + split_manager = Splitter(splitter_config) for split in split_manager.split_iterator(): logger.info(f"Training split {split}") @@ -244,7 +244,7 @@ def train( def check_split_list(self, split_list, overwrite): existing_splits = [] splitter_config = SplitterConfig(**self.config.get_dict()) - split_manager = Splitter(splitter_config, split_list=split_list) + split_manager = Splitter(splitter_config) for split in split_manager.split_iterator(): split_path = self.maps_manager.maps_path / f"split-{split}" if split_path.is_dir(): @@ -281,7 +281,7 @@ def _resume( """ missing_splits = [] splitter_config = SplitterConfig(**self.config.get_dict()) - split_manager = Splitter(splitter_config, split_list=split_list) + split_manager = Splitter(splitter_config) for split in split_manager.split_iterator(): if not (self.maps_manager.maps_path / f"split-{split}" / "tmp").is_dir(): missing_splits.append(split) @@ -473,8 +473,8 @@ def _train_ssda( splitter_config = SplitterConfig(**self.config.get_dict()) - split_manager = Splitter(splitter_config, split_list=split_list) - split_manager_target_lab = Splitter(splitter_config, split_list=split_list) + split_manager = Splitter(splitter_config) + split_manager_target_lab = Splitter(splitter_config) for split in split_manager.split_iterator(): logger.info(f"Training split {split}")