From 67a8644c9663356ee1a0145e5bc3e1a4cf727be9 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 16 Jan 2020 17:19:40 -0500 Subject: [PATCH] added loggers --- pytorch_lightning/logging/__init__.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pytorch_lightning/logging/__init__.py b/pytorch_lightning/logging/__init__.py index be48000d..1bb34759 100644 --- a/pytorch_lightning/logging/__init__.py +++ b/pytorch_lightning/logging/__init__.py @@ -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')