From 3a56a1f00d0a314d1ccc9ca0719b5d40123d3416 Mon Sep 17 00:00:00 2001
From: shippingwang <shipeng1108@163.com>
Date: Tue, 9 Jun 2020 07:22:11 +0000
Subject: [PATCH] hide visualdl

---
 tools/train.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/tools/train.py b/tools/train.py
index 4384b152..a5f765f0 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -19,7 +19,6 @@ from __future__ import print_function
 import argparse
 import os
 
-from visualdl import LogWriter
 import paddle.fluid as fluid
 from paddle.fluid.incubate.fleet.base import role_maker
 from paddle.fluid.incubate.fleet.collective import fleet
@@ -101,7 +100,12 @@ def main(args):
         compiled_valid_prog = program.compile(config, valid_prog)
 
     compiled_train_prog = fleet.main_program
-    vdl_writer = LogWriter(args.vdl_dir) if args.vdl_dir else None
+
+    if args.vdl_dir:
+        from visualdl import LogWriter
+        vdl_writer = LogWriter(args.vdl_dir)
+    else:
+        vdl_writer = None
 
     for epoch_id in range(config.epochs):
         # 1. train with train dataset
-- 
GitLab