Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update tensorboard.py (#3920)

This commit is contained in:
ananthsub 2020-10-06 17:34:15 -07:00 committed by GitHub
parent 4cd14c4237
commit 6c47535de8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch_lightning/loggers/tensorboard.py
View File

@ -221,6 +221,7 @@ class TensorBoardLogger(LightningLoggerBase):
@rank_zero_only @rank_zero_only
def finalize(self, status: str) -> None: def finalize(self, status: str) -> None:
self.experiment.flush()
self.save() self.save()
@property @property