@@ -48,7 +48,6 @@ def main(
4848 device = torch .device ("cuda" if torch .cuda .is_available () else "cpu" )
4949 log (INFO , f"Using device: { device } " )
5050 log (INFO , f"Using server address: { server_address } " )
51- log (INFO , f"nnUNet_raw: { nnUNet_raw } " )
5251
5352 # Load the dataset if necessary
5453 msd_dataset_enum = get_msd_dataset_enum (msd_dataset_name )
@@ -59,9 +58,7 @@ def main(
5958
6059 # The dataset ID will be the same as the MSD Task number
6160 dataset_id = int (msd_dataset_enum .value [4 :6 ])
62- nnunet_dataset_name = (
63- f"Dataset{ dataset_id :03d} _{ msd_dataset_enum .value .split ('_' )[1 ]} "
64- )
61+ nnunet_dataset_name = f"Dataset{ dataset_id :03d} _{ msd_dataset_enum .value .split ('_' )[1 ]} "
6562
6663 # Convert the msd dataset if necessary
6764 if not exists (join (nn_unet_raw , nnunet_dataset_name )):
@@ -87,9 +84,7 @@ def main(
8784 # State checkpointer (being overhauled soon)
8885 if intermediate_client_state_dir is not None :
8986 checkpoint_and_state_module = ClientCheckpointAndStateModule (
90- state_checkpointer = ClientStateCheckpointer (
91- Path (intermediate_client_state_dir )
92- )
87+ state_checkpointer = ClientStateCheckpointer (Path (intermediate_client_state_dir ))
9388 )
9489 else :
9590 checkpoint_and_state_module = None
@@ -215,10 +210,6 @@ def main(
215210 args = parser .parse_args ()
216211
217212 # Set the random seed for reproducibility
218- < << << << HEAD
219- set_all_random_seeds (
220- args .seed , disable_torch_benchmarking = True , use_deterministic_torch_algos = True
221- == == == =
222213
223214 # NOTE: This implementation does not cover all sources of randomness in nnUNet, so complete
224215 # determinism cannot be achieved. The nnUNet maintainers have confirmed that full determinism
@@ -231,7 +222,6 @@ def main(
231222 args .seed ,
232223 disable_torch_benchmarking = True ,
233224 use_deterministic_torch_algos = True ,
234- >> >> >> > b9256b2d (cr )
235225 )
236226
237227 # Set the log level
0 commit comments