diff --git a/pytorch_lightning/trainer/callback_config.py b/pytorch_lightning/trainer/callback_config.py index 02db09c3..1f23d3c1 100644 --- a/pytorch_lightning/trainer/callback_config.py +++ b/pytorch_lightning/trainer/callback_config.py @@ -2,7 +2,7 @@ import os from abc import ABC from pytorch_lightning.callbacks import ModelCheckpoint, EarlyStopping -from pytorch_lightning.logging import TestTubeLogger +from pytorch_lightning.logging import TensorboardLogger class TrainerCallbackConfigMixin(ABC): @@ -69,7 +69,7 @@ class TrainerCallbackConfigMixin(ABC): # configure logger if logger is True: # default logger - self.logger = TestTubeLogger( + self.logger = TensorboardLogger( save_dir=self.default_save_path, version=self.slurm_job_id, name='lightning_logs' diff --git a/requirements.txt b/requirements.txt index 6cafa389..c6dc2c75 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,5 +4,4 @@ numpy>=1.16.4 torch>=1.1 torchvision>=0.4.0 pandas>=0.24 # lower version do not support py3.7 -test-tube>=0.7.5 future>=0.17.1 # required for buildins in setup.py