diff --git a/nnunetv2/inference/predict_from_raw_data.py b/nnunetv2/inference/predict_from_raw_data.py index 3e3e3dd56..1fe69ea1b 100644 --- a/nnunetv2/inference/predict_from_raw_data.py +++ b/nnunetv2/inference/predict_from_raw_data.py @@ -607,7 +607,7 @@ def predict_sliding_window_return_logits(self, input_image: torch.Tensor) \ device=results_device) if self.use_gaussian: gaussian = compute_gaussian(tuple(self.configuration_manager.patch_size), sigma_scale=1. / 8, - value_scaling_factor=1000, + value_scaling_factor=10, device=results_device) except RuntimeError: # sometimes the stuff is too large for GPUs. In that case fall back to CPU @@ -620,7 +620,7 @@ def predict_sliding_window_return_logits(self, input_image: torch.Tensor) \ device=results_device) if self.use_gaussian: gaussian = compute_gaussian(tuple(self.configuration_manager.patch_size), sigma_scale=1. / 8, - value_scaling_factor=1000, + value_scaling_factor=10, device=results_device) finally: empty_cache(self.device)