mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-10 12:21:57 +08:00
added loggers
This commit is contained in:
@@ -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')
|
||||
|
||||
Reference in New Issue
Block a user