Skip to content

Commit

Permalink
Remove rank_zero_only decorators from model checkpoint entry points
Browse files Browse the repository at this point in the history
  • Loading branch information
ananthsub committed Sep 19, 2020
1 parent 978574e commit 4ebfa42
Showing 1 changed file with 0 additions and 15 deletions.
15 changes: 0 additions & 15 deletions pytorch_lightning/callbacks/model_checkpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -416,16 +416,7 @@ def on_validation_end(self, trainer, pl_module):
self._del_model(self.last_model_path)

def _do_check_save(
<<<<<<< HEAD
self, filepath: str, current: torch.Tensor, epoch: int, trainer, pl_module
=======
self,
filepath: str,
current: torch.Tensor,
epoch: int,
trainer,
pl_module,
>>>>>>> Update model_checkpoint.py
):
# remove kth

Expand Down Expand Up @@ -461,13 +452,7 @@ def _do_check_save(
if cur_path != filepath:
self._del_model(cur_path)

<<<<<<< HEAD
def on_save_checkpoint(self, trainer, pl_module) -> Dict[str, Any]:
=======
def on_save_checkpoint(
self, trainer, pl_module
) -> Dict[str, Any]:
>>>>>>> Update model_checkpoint.py
return {
"best_model_score": self.best_model_score,
"best_model_path": self.best_model_path,
Expand Down

0 comments on commit 4ebfa42

Please sign in to comment.