mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
made tensorboard the default not test-tube
This commit is contained in:
@@ -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'
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user