Skip to content

Commit 75a99eb

Browse files
committed
hanging rebase conflict
1 parent 2eadc17 commit 75a99eb

1 file changed

Lines changed: 2 additions & 12 deletions

File tree

examples/nnunet_example/client.py

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)