From 2a15e391669f9073ba10ef7ff20bb54ec5246977 Mon Sep 17 00:00:00 2001
From: erogol <erogol@hotmail.com>
Date: Tue, 10 Mar 2020 22:38:51 +0100
Subject: [PATCH] bug fix and run desc in tensorboard

---
 config.json     | 4 ++--
 train.py        | 7 +++++--
 utils/logger.py | 3 +++
 3 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/config.json b/config.json
index e525ec31..efc96c9e 100644
--- a/config.json
+++ b/config.json
@@ -1,7 +1,7 @@
 {
     "model": "Tacotron2",          // one of the model in models/  
-    "run_name": "ljspeech-stft_params",
-    "run_description": "tacotron2 cosntant stf parameters",
+    "run_name": "ljspeech",
+    "run_description": "tacotron2 with guided attention and -1 1 normalization and no preemphasis",
 
     // AUDIO PARAMETERS
     "audio":{
diff --git a/train.py b/train.py
index 15c65f64..ea6d391c 100644
--- a/train.py
+++ b/train.py
@@ -47,7 +47,7 @@ def setup_loader(ap, r, is_val=False, verbose=False):
         dataset = MyDataset(
             r,
             c.text_cleaner,
-            compute_linear_spec=True if c.model.lower() is 'tacotron' else False
+            compute_linear_spec=True if c.model.lower() is 'tacotron' else False,
             meta_data=meta_data_eval if is_val else meta_data_train,
             ap=ap,
             tp=c.characters if 'characters' in c.keys() else None,
@@ -410,7 +410,7 @@ def evaluate(model, criterion, ap, global_step, epoch):
                              loss_dict['ga_loss'].item(),
                              keep_avg['avg_ga_loss'],
                              align_score, keep_avg['avg_align_score']),
-                    flush=Tr ue)
+                    flush=True)
 
         if args.rank == 0:
             # Diagnostic visualizations
@@ -696,6 +696,9 @@ if __name__ == '__main__':
         LOG_DIR = OUT_PATH
         tb_logger = Logger(LOG_DIR)
 
+    # write model desc to tensorboard
+    tb_logger.tb_add_text('model-description', c['run_description'], 0)
+
     try:
         main(args)
     except KeyboardInterrupt:
diff --git a/utils/logger.py b/utils/logger.py
index 51a10422..e5faeda4 100644
--- a/utils/logger.py
+++ b/utils/logger.py
@@ -75,3 +75,6 @@ class Logger(object):
 
     def tb_test_figures(self, step, figures):
         self.dict_to_tb_figure("TestFigures", figures, step)
+
+    def tb_add_text(self, title, text, step):
+        self.writer.add_text(title, text, step)