2016-08-19 34 views
0

Tensorboard可以通过将每个运行存储在日志目录的子目录中来可视化张量流图的几次运行。 例如,the documentation提供这个例子:张量流可以自动创建一个独特的运行目录吗?

experiments/ 
experiments/run1/ 
experiments/run1/events.out.tfevents.1456525581.name 
experiments/run1/events.out.tfevents.1456525585.name 
experiments/run2/ 
experiments/run2/events.out.tfevents.1456525385.name 
/tensorboard --logdir=experiments 

开始下运行(RUN3),一个新的目录应该被传递到SummaryWriter构造:

summary_writer = tf.train.SummaryWriter('experiments/run3/', sess.graph) 

所在目录是顶级级别的日志目录(实验)和唯一的ID(run3)。

有没有办法自动创建一个新的唯一运行ID? 顺序整数ID会很好,基于时间的ID也是一样。

回答

2

您可以在python中检查experiments中存在的目录,并用增加的数字创建一个新目录。

如果列表为空,我们从run_01开始。

import os 

previous_runs = os.listdir('experiments') 
if len(previous_runs) == 0: 
    run_number = 1 
else: 
    run_number = max([int(s.split('run_')[1]) for s in previous_runs]) + 1 

logdir = 'run_%02d' % run_number 
summary_writer = tf.train.SummaryWriter(os.path.join('experiments', logdir), sess.graph) 

我用"%02d"有类似名称:run_01, run_02, run_03, ... run_10, run_11

相关问题