From e5de29a954a6873872b256dc0dc077a3ffe069f8 Mon Sep 17 00:00:00 2001 From: Joshua Meier Date: Wed, 29 Aug 2018 15:25:47 -0700 Subject: [PATCH] instructions for tensorboard (#61) --- baselines/common/tf_util.py | 20 +++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) diff --git a/baselines/common/tf_util.py b/baselines/common/tf_util.py index 6a71e1f..3da9441 100644 --- a/baselines/common/tf_util.py +++ b/baselines/common/tf_util.py @@ -398,6 +398,20 @@ def _squeeze_shape(shape): # ================================================================ def launch_tensorboard_in_background(log_dir): - import subprocess - print(log_dir) - p = subprocess.Popen(['tensorboard', '--logdir', log_dir]) + ''' + To log the Tensorflow graph when using rl-algs + algorithms, you can run the following code + in your main script: + import threading, time + def start_tensorboard(session): + time.sleep(10) # Wait until graph is setup + tb_path = osp.join(logger.get_dir(), 'tb') + summary_writer = tf.summary.FileWriter(tb_path, graph=session.graph) + summary_op = tf.summary.merge_all() + launch_tensorboard_in_background(tb_path) + session = tf.get_default_session() + t = threading.Thread(target=start_tensorboard, args=([session])) + t.start() + ''' + import subprocess + subprocess.Popen(['tensorboard', '--logdir', log_dir])