made tensorboard the default not test-tube

This commit is contained in:
William Falcon
2019-12-07 23:36:51 -05:00
parent 47a82cf1b9
commit 99c9b82527
2 changed files with 2 additions and 3 deletions
+2 -2
View File
@@ -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'
-1
View File
@@ -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