diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index b4d2b31..d31891f 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -659,7 +659,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... random_ct_samples_path=training_data_dst_path if ct_mode is not None and not self.pretrain else None - cpu_count = min(multiprocessing.cpu_count(), 8) + cpu_count = multiprocessing.cpu_count() src_generators_count = cpu_count // 2 dst_generators_count = cpu_count // 2 if ct_mode is not None: