From abf52a88b07d1b8e43ddaa919b097b091f13ed69 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Thu, 7 Feb 2019 17:29:37 +0100 Subject: [PATCH] Properly created global step 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 7e55532..6697c1b 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -99,7 +99,7 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i y_real_z = k.ones(batch_size) y_fake_z = k.zeros(batch_size) sample = k.expand_dims(k.expand_dims(k.random_normal((64, zsize)), axis=1), axis=1) - global_step = k.variable(0) + global_step = tf.train.get_or_create_global_step() encoder_loss_history = [] decoder_loss_history = []