Save tensorboard logs to a model subdir

eae63621 · Matt Harvey · 040284d9 · eae63621
Commit eae63621 authored 7 years ago by Matt Harvey
Hide whitespace changes
Inline Side-by-side

Showing

with 1 addition and 1 deletion
+1 -1
--- a/train.py
+++ b/train.py
@@ -18,7 +18,7 @@ def train(data_type, seq_length, model, saved_model=None,
        save_best_only=True)

    # Helper: TensorBoard
-    tb = TensorBoard(log_dir=os.path.join('data', 'logs'))
+    tb = TensorBoard(log_dir=os.path.join('data', 'logs', model))

    # Helper: Stop when we stop learning.
    early_stopper = EarlyStopping(patience=5)