Add VisualDL to replace tb. (#509)

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):

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)

requirements.txt

@@ -4,3 +4,4 @@ six==1.11.0
 sklearn==0.0
 sentencepiece==0.1.8
 jieba==0.39
+visualdl>=2.0.0b7