Update tensorboard.py (#3920)

2020-10-06 17:34:15 -07:00 · 2020-10-06 17:34:15 -07:00 · 6c47535de8
parent 4cd14c4237
commit 6c47535de8
1 changed files with 1 additions and 0 deletions
--- a/pytorch_lightning/loggers/tensorboard.py
+++ b/pytorch_lightning/loggers/tensorboard.py
@ -221,6 +221,7 @@ class TensorBoardLogger(LightningLoggerBase):

    @rank_zero_only
    def finalize(self, status: str) -> None:
+        self.experiment.flush()
        self.save()

    @property