From d86c53c18edb09c0d91b89244214043801f411a7 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 15 Apr 2019 17:10:48 +0200 Subject: [PATCH] Save reconstruction image summaries only with verbose on Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index 4c8d0e9..8783e32 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -169,7 +169,7 @@ def _train_one_epoch_simple(epoch: int, global_step=global_step) enc_dec_loss_avg(reconstruction_loss) - if int(global_step % LOG_FREQUENCY) == 0: + if int(global_step % LOG_FREQUENCY) == 0 and verbose: comparison = K.concatenate([x[:int(batch_size / 2)], x_decoded[:int(batch_size / 2)]], axis=0) grid = util.prepare_image(comparison.cpu(), nrow=int(batch_size/2)) summary_ops_v2.image(name='reconstruction',