1. 程式人生 > >tensorflow tensorboard使用

tensorflow tensorboard使用

構建計算圖

#構建計算圖的過程略去
tf.summary.scalar('train_loss', train_loss)
tf.summary.scalar('train_acc', train_acc)
tf.summary.scalar('eval_loss', eval_loss)
tf.summary.scalar('eval_acc', eval_acc)
summary_op = tf.summary.merge_all()

sess = tf.Session()
writer = tf.summary.FileWriter(config.log_path, sess.graph)
for step in range(max_step):
	summary_str = sess.run(summary_op)
	writer.add_summary(summary_str, step)

如何連線遠端伺服器到本機顯示

檢視tensorboard,登入使得6006埠與本機16006埠相連:登入命令:

ssh -L 16006:127.0.0.1:6006 [email protected]###.###.### -p 22 
python3.4 /usr/local/lib/python3.4/distpackages/tensorflow/tensorboard/tensorboard.py --logdir=/storage/log/