Skip to content

Commit

Permalink
typos
Browse files Browse the repository at this point in the history
  • Loading branch information
FabianIsensee committed Feb 8, 2024
1 parent d2bd77d commit 997804c
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion nnunetv2/inference/predict_from_raw_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -469,7 +469,7 @@ def predict_logits_from_preprocessed_data(self, data: torch.Tensor) -> torch.Ten
self.network._orig_mod.load_state_dict(params)

# why not leave prediction on device if perform_everything_on_device? Because this may cause the
# second iteration to crash due to OOM. Grabbing tha twith try except cause way more bloated code than
# second iteration to crash due to OOM. Grabbing that with try except cause way more bloated code than
# this actually saves computation time
if prediction is None:
prediction = self.predict_sliding_window_return_logits(data).to('cpu')
Expand Down
2 changes: 1 addition & 1 deletion nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -1241,7 +1241,7 @@ def perform_actual_validation(self, save_probabilities: bool = False):
self.dataset_json),
)
))
# if we don't barrier from time to time we will get nccl timeouts for large datsets. Yuck.
# if we don't barrier from time to time we will get nccl timeouts for large datasets. Yuck.
if self.is_ddp and i < last_barrier_at_idx and (i + 1) % 20 == 0:
dist.barrier()

Expand Down

0 comments on commit 997804c

Please sign in to comment.