wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-11 12:31:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

cleaned up test_tube logger

Browse Source
This commit is contained in:
William Falcon
2019-10-05 23:23:38 -04:00
parent 4e1c90d892
commit d0a2438e22
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/logging/test_tube_logger.py
+4 -1
View File
@@ -10,11 +10,13 @@ class TestTubeLogger(LightningLoggerBase):
__test__ = False
def __init__(
self, save_dir, name="default", debug=False, version=None, create_git_tag=False
self, save_dir, name="default", description=None, debug=False,
version=None, create_git_tag=False
):
super().__init__()
self.save_dir = save_dir
self.name = name
self.description = description
self.debug = debug
self._version = version
self.create_git_tag = create_git_tag
@@ -29,6 +31,7 @@ class TestTubeLogger(LightningLoggerBase):
name=self.name,
debug=self.debug,
version=self.version,
description=self.description,
create_git_tag=self.create_git_tag,
rank=self.rank,
)