From f81ecbac0d93d4531868fc85168e88fec7b639ef Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 10 Apr 2019 15:16:13 +0200 Subject: [PATCH] Fixed slicing 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 d6b585c..f6bdc2f 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -171,7 +171,7 @@ def _train_one_epoch_simple(epoch: int, enc_dec_loss_avg(reconstruction_loss) if int(global_step % LOG_FREQUENCY) == 0: - comparison = K.concatenate([x[:batch_size/2], x_decoded[:batch_size/2]], axis=0) + 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', tensor=K.expand_dims(grid, axis=0), max_images=1,