From 97bf461cf9c00d182b0cc841c6b966a0ca9e85a4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Thu, 5 Nov 2020 01:45:02 +0100 Subject: [PATCH] trigger ci --- pytorch_lightning/callbacks/model_checkpoint.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/callbacks/model_checkpoint.py b/pytorch_lightning/callbacks/model_checkpoint.py index 413e96a39a..2f2bce91fd 100644 --- a/pytorch_lightning/callbacks/model_checkpoint.py +++ b/pytorch_lightning/callbacks/model_checkpoint.py @@ -200,6 +200,7 @@ class ModelCheckpoint(Callback): This method runs on all ranks, it is the responsibility of `self.save_function` to handle correct behaviour in distributed training, i.e., saving only on rank 0. """ + epoch = trainer.current_epoch global_step = trainer.global_step