From fc1aa203968717ec0463dceb21f1416007e64305 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 8 Feb 2019 11:37:35 +0100 Subject: [PATCH] Improved saving location of summaries Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index 13c7ed4..284eee7 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -399,6 +399,9 @@ def extract_batch(data: np.ndarray, it: int, batch_size: int) -> tfe.Variable: if __name__ == "__main__": tf.enable_eager_execution() - train_summary_writer = summary_ops_v2.create_file_writer('./summaries/train') + inlier_classes = [0] + iteration = 1 + train_summary_writer = summary_ops_v2.create_file_writer( + './summaries/train/number-' + str(inlier_classes[0]) + '/' + str(iteration)) with train_summary_writer.as_default(), summary_ops_v2.always_record_summaries(): - train_mnist(folding_id=0, inlier_classes=[0], total_classes=10) + train_mnist(folding_id=0, inlier_classes=inlier_classes, total_classes=10)