Added summary writer

Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 08:39:59 +01:00
parent ab52f0feb0
commit 2112e1dd0c
1 changed files with 3 additions and 1 deletions
--- a/src/twomartens/masterthesis/aae/train.py
+++ b/src/twomartens/masterthesis/aae/train.py
@ -400,4 +400,6 @@ def extract_batch(data: np.ndarray, it: int, batch_size: int) -> tfe.Variable:

 if __name__ == "__main__":
    tf.enable_eager_execution()
-    train_mnist(folding_id=0, inlier_classes=[0], total_classes=10)
+    train_summary_writer = summary_ops_v2.create_file_writer('./summaries/train')
+    with train_summary_writer.as_default(), summary_ops_v2.always_record_summaries():
+        train_mnist(folding_id=0, inlier_classes=[0], total_classes=10)