provide checkpoint path

hasan7n · Oct 21, 2024 · 7314eef · 7314eef
1 parent e774fe9
commit 7314eef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/fl_post/fl/project/src/runner_nnunetv1.py b/examples/fl_post/fl/project/src/runner_nnunetv1.py
@@ -124,7 +124,7 @@ def write_tensors_into_checkpoint(self, tensor_dict, with_opt_vars):
         # get device for correct placement of tensors
         device = self.device
 
-        checkpoint_dict = self.load_checkpoint(map_location=device)
+        checkpoint_dict = self.load_checkpoint(checkpoint_path=self.checkpoint_path_load, map_location=device)
         epoch = checkpoint_dict['epoch']
         new_state = {}
         # grabbing keys from the checkpoint state dict, poping from the tensor_dict