From be40a52dde101919a529fed7e035af847fe193cb Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 17 Apr 2019 13:51:48 +0200 Subject: [PATCH] Fixed nested arguments Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/run.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/aae/run.py b/src/twomartens/masterthesis/aae/run.py index 72dcb6d..ffa49d0 100644 --- a/src/twomartens/masterthesis/aae/run.py +++ b/src/twomartens/masterthesis/aae/run.py @@ -109,8 +109,8 @@ def _run_one_epoch_simple(dataset: tf.data.Dataset, enc_dec_loss_avg(reconstruction_loss) if int(global_step % train.LOG_FREQUENCY) == 0 and debug: - comparison = K.concatenate([x[:int(batch_size / 2)], x_decoded[:int(batch_size / 2), - z[:int(batch_size / 2)]]], axis=0) + comparison = K.concatenate([x[:int(batch_size / 2)], x_decoded[:int(batch_size / 2)], + z[:int(batch_size / 2)]], axis=0) grid = util.prepare_image(comparison.cpu(), nrow=int(batch_size / 2)) summary_ops_v2.image(name='reconstruction', tensor=K.expand_dims(grid, axis=0), max_images=1,