set weights_only=False in torch.load

MIC-DKFZ · Jan 30, 2025 · dadc00d · dadc00d
1 parent 3df680c
commit dadc00d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nnunetv2/inference/predict_from_raw_data.py b/nnunetv2/inference/predict_from_raw_data.py
@@ -82,7 +82,7 @@ def initialize_from_trained_model_folder(self, model_training_output_dir: str,
         for i, f in enumerate(use_folds):
             f = int(f) if f != 'all' else f
             checkpoint = torch.load(join(model_training_output_dir, f'fold_{f}', checkpoint_name),
-                                    map_location=torch.device('cpu'))
+                                    map_location=torch.device('cpu'), weights_only=False)
             if i == 0:
                 trainer_name = checkpoint['trainer_name']
                 configuration_name = checkpoint['init_args']['configuration']