added loggers

This commit is contained in:
William Falcon
2020-01-16 17:19:40 -05:00
parent 275ebdcc51
commit 67a8644c96
+3 -1
View File
@@ -77,7 +77,7 @@ from os import environ
from .base import LightningLoggerBase, rank_zero_only
from .tensorboard import TensorBoardLogger
all = ['TensorBoardLogger']
all = []
try:
# needed to prevent ImportError and duplicated logs.
@@ -100,6 +100,8 @@ try:
except ImportError:
pass
all.append('TensorBoardLogger')
try:
from .test_tube import TestTubeLogger
all.append('TestTubeLogger')