gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

instructions for tensorboard (#61)

Browse Source
This commit is contained in:
Joshua Meier
2018-08-29 15:25:47 -07:00
committed by Peter Zhokhov
parent 2507d335f9
commit e5de29a954
1 changed files with 17 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
baselines/common/tf_util.py
View File

@@ -398,6 +398,20 @@ def _squeeze_shape(shape):
# ================================================================ # ================================================================
def launch_tensorboard_in_background(log_dir): def launch_tensorboard_in_background(log_dir):
import subprocess '''
print(log_dir) To log the Tensorflow graph when using rl-algs
p = subprocess.Popen(['tensorboard', '--logdir', log_dir]) 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])