Update predict_from_raw_data.py

adrianvallsc · web-flow · commit b1df4eb590ae · 2024-06-05T00:50:50.000+02:00
Error from RuntimeError: Inplace update to inference tensor outside InferenceMode is not allowed.You can make a clone to get a normal tensor before doing inplace update.See pytorch/rfcs#17 for more details
diff --git a/nnunetv2/inference/predict_from_raw_data.py b/nnunetv2/inference/predict_from_raw_data.py
@@ -491,7 +491,7 @@ def predict_logits_from_preprocessed_data(self, data: torch.Tensor) -> torch.Ten
             if prediction is None:
                 prediction = self.predict_sliding_window_return_logits(data).to('cpu')
             else:
-                prediction += self.predict_sliding_window_return_logits(data).to('cpu')
+                prediction = prediction + self.predict_sliding_window_return_logits(data).to('cpu')
 
         if len(self.list_of_parameters) > 1:
             prediction /= len(self.list_of_parameters)
