update logger init (#727)

* update logger init

* formatting
This commit is contained in:
Jirka Borovec
2020-01-23 11:36:40 -05:00
committed by William Falcon
parent 50881c0b31
commit a804755e6e
2 changed files with 9 additions and 10 deletions
+7 -9
View File
@@ -76,41 +76,39 @@ from os import environ
from .base import LightningLoggerBase, rank_zero_only
from .tensorboard import TensorBoardLogger
all = []
loggers = ['TensorBoardLogger']
try:
# needed to prevent ImportError and duplicated logs.
environ["COMET_DISABLE_AUTO_LOGGING"] = "1"
from .comet import CometLogger
all.append('CometLogger')
loggers.append('CometLogger')
except ImportError:
del environ["COMET_DISABLE_AUTO_LOGGING"]
try:
from .mlflow import MLFlowLogger
all.append('MLFlowLogger')
loggers.append('MLFlowLogger')
except ImportError:
pass
try:
from .neptune import NeptuneLogger
all.append('NeptuneLogger')
loggers.append('NeptuneLogger')
except ImportError:
pass
all.append('TensorBoardLogger')
try:
from .test_tube import TestTubeLogger
all.append('TestTubeLogger')
loggers.append('TestTubeLogger')
except ImportError:
pass
try:
from .wandb import WandbLogger
all.append('WandbLogger')
loggers.append('WandbLogger')
except ImportError:
pass
__all__ = all
__all__ = loggers
+2 -1
View File
@@ -8,4 +8,5 @@ warnings.warn("`root_module` package has been renamed to `core` since v0.6.0"
" and will be removed in v0.8.0", DeprecationWarning)
from pytorch_lightning.core import ( # noqa: E402
decorators, grads, hooks, root_module, memory, model_saving)
decorators, grads, hooks, root_module, memory, model_saving
)