From 7314eef8328ccbe2bb8615687fab1041d5e3c37f Mon Sep 17 00:00:00 2001 From: "Edwards, Brandon" Date: Mon, 21 Oct 2024 14:06:30 -0700 Subject: [PATCH] provide checkpoint path --- examples/fl_post/fl/project/src/runner_nnunetv1.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/fl_post/fl/project/src/runner_nnunetv1.py b/examples/fl_post/fl/project/src/runner_nnunetv1.py index b2387e55b..7a9c9039a 100644 --- 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