Add --no_tensorboard flags to disable Tensorboard summaries

cangermueller · Oct 24, 2017 · 0332e85 · 0332e85
1 parent 3db804c
commit 0332e85
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scripts/dcpg_train.py b/scripts/dcpg_train.py
@@ -324,6 +324,10 @@ def create_parser(self, name):
             help='L2 weight decay',
             type=float,
             default=0.0001)
+        g.add_argument(
+            '--no_tensorboard',
+            help='Do not store Tensorboard summaries',
+            action='store_true')
 
         g = p.add_argument_group('arguments to select outputs and weights')
         g.add_argument(
@@ -449,7 +453,7 @@ def save_lc(epoch, epoch_logs, val_epoch_logs):
         )
         callbacks.append(self.perf_logger)
 
-        if K._BACKEND == 'tensorflow':
+        if K._BACKEND == 'tensorflow' and not opts.no_tensorboard:
             callbacks.append(kcbk.TensorBoard(
                 log_dir=opts.out_dir,
                 histogram_freq=1,