diff --git a/propeller/paddle/train/hooks.py b/propeller/paddle/train/hooks.py index 9e4f45774234209a9a3053c8e5dc3fd752553f15..e3820a7bb25affb3925b9ed6939658256329a391 100644 --- a/propeller/paddle/train/hooks.py +++ b/propeller/paddle/train/hooks.py @@ -144,7 +144,7 @@ class TqdmNotebookProgressBarHook(RunHook): class LoggingHook(RunHook): - """log tensor in to screan and tensorboard""" + """log tensor in to screan and VisualDL""" def __init__(self, loss, @@ -205,7 +205,7 @@ class LoggingHook(RunHook): speed = -1. self.last_state = state - # log to tensorboard + # log to VisualDL if self.writer is not None: self.writer.add_scalar('loss', loss, state.gstep) for name, t in zip(self.s_name, s_np): diff --git a/propeller/paddle/train/trainer.py b/propeller/paddle/train/trainer.py index 625c8666bb0e8a96182de0501cea786b39361ac9..a9eab323bc584df000636ceef17c35ac795df723 100644 --- a/propeller/paddle/train/trainer.py +++ b/propeller/paddle/train/trainer.py @@ -48,11 +48,11 @@ __all__ = ['train_and_eval', 'Learner'] def _get_summary_writer(path): summary_writer = None try: - from tensorboardX import SummaryWriter + from visualdl import LogWriter if distribution.status.is_master: - summary_writer = SummaryWriter(os.path.join(path)) + summary_writer = LogWriter(os.path.join(path)) except ImportError: - log.warning('tensorboardX not installed, will not log to tensorboard') + log.warning('VisualDL not installed, will not log to VisualDL') return summary_writer @@ -69,7 +69,7 @@ def _log_eval_result(name, eval_result, swriter, state): printable.append('{}\t{}'.format(n, val)) if swriter is not None: swriter.add_scalar(n, val, state.gstep) - log.debug('write to tensorboard %s' % swriter.logdir) + log.debug('write to VisualDL %s' % swriter.logdir) if len(printable): log.info('*** eval res: %10s ***' % name) diff --git a/requirements.txt b/requirements.txt index cacf671c0f65f21a48ada5396afce164c10d1437..b3d835609d688f8c031d18f10249a64411d36701 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,3 +4,4 @@ six==1.11.0 sklearn==0.0 sentencepiece==0.1.8 jieba==0.39 +visualdl>=2.0.0b7