Add tensorboard export script

2018-08-13 12:38:22 +02:00 · 2018-08-13 12:38:22 +02:00 · fd623c32de
parent 0c07241104
commit fd623c32de
1 changed files with 11 additions and 0 deletions
--- a/pywatts/board.py
+++ b/pywatts/board.py
@ -0,0 +1,11 @@
+import tensorflow as tf
+import subprocess
+
+writer = tf.summary.FileWriter("tensorboard")
+checkpoint = tf.train.get_checkpoint_state('tf_pywatts_model')
+with tf.Session() as sess:
+    saver = tf.train.import_meta_graph(checkpoint.model_checkpoint_path + '.meta')
+    saver.restore(sess, checkpoint.model_checkpoint_path)
+writer.add_graph(sess.graph)
+
+subprocess.check_output(['tensorboard', '--logdir', 'tensorboard'])