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