From b00b41f240540bac12911519c05fcc8012e594d7 Mon Sep 17 00:00:00 2001 From: Fabian Isensee Date: Tue, 12 Dec 2023 14:29:17 +0100 Subject: [PATCH] bugfix --- nnunetv2/inference/examples.py | 2 +- nnunetv2/inference/readme.md | 2 +- nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/nnunetv2/inference/examples.py b/nnunetv2/inference/examples.py index b57a39831..a66d98f8b 100644 --- a/nnunetv2/inference/examples.py +++ b/nnunetv2/inference/examples.py @@ -12,7 +12,7 @@ tile_step_size=0.5, use_gaussian=True, use_mirroring=True, - perform_everything_on_gpu=True, + perform_everything_on_device=True, device=torch.device('cuda', 0), verbose=False, verbose_preprocessing=False, diff --git a/nnunetv2/inference/readme.md b/nnunetv2/inference/readme.md index 721952888..4f832a158 100644 --- a/nnunetv2/inference/readme.md +++ b/nnunetv2/inference/readme.md @@ -57,7 +57,7 @@ Example: tile_step_size=0.5, use_gaussian=True, use_mirroring=True, - perform_everything_on_gpu=True, + perform_everything_on_device=True, device=torch.device('cuda', 0), verbose=False, verbose_preprocessing=False, diff --git a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py index 27439b170..821a4e00f 100644 --- a/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py +++ b/nnunetv2/training/nnUNetTrainer/nnUNetTrainer.py @@ -1122,7 +1122,7 @@ def perform_actual_validation(self, save_probabilities: bool = False): self.network.eval() predictor = nnUNetPredictor(tile_step_size=0.5, use_gaussian=True, use_mirroring=True, - perform_everything_on_gpu=True, device=self.device, verbose=False, + perform_everything_on_device=True, device=self.device, verbose=False, verbose_preprocessing=False, allow_tqdm=False) predictor.manual_initialization(self.network, self.plans_manager, self.configuration_manager, None, self.dataset_json, self.__class__.__name__,