From 9623fc43bec2188d56a28593628951be2a2f3fe8 Mon Sep 17 00:00:00 2001
From: ShenYuhan <yhan_shen@163.com>
Date: Mon, 6 Jul 2020 10:56:08 +0800
Subject: [PATCH] Add VisualDL to replace tb. (#509)

---
 propeller/paddle/train/hooks.py   | 4 ++--
 propeller/paddle/train/trainer.py | 8 ++++----
 requirements.txt                  | 1 +
 3 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/propeller/paddle/train/hooks.py b/propeller/paddle/train/hooks.py
index 9e4f457..e3820a7 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 625c866..a9eab32 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 cacf671..b3d8356 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
-- 
GitLab