Update substrafl/algorithms/pytorch/torch_base_algo.py

Substra · Jan 19, 2024 · a7172fe · a7172fe
1 parent 5100f95
commit a7172fe
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/substrafl/algorithms/pytorch/torch_base_algo.py b/substrafl/algorithms/pytorch/torch_base_algo.py
@@ -265,8 +265,6 @@ def _update_from_checkpoint(self, checkpoint: dict) -> None:
                     self._strategy_specific_variable = checkpoint.pop("strategy_specific_variable")
                     return
         """
-        # assert path.is_file(), f'Cannot load the model - does not exist {list(path.parent.glob("*"))}'
-        # checkpoint = torch.load(path, map_location=self._device)
         self._model.load_state_dict(checkpoint.pop("model_state_dict"))
 
         if self._optimizer is not None: