From 2e076597ef301cb1440396e88830d1d1dc008528 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 15 Apr 2019 12:16:38 +0200 Subject: [PATCH] Fixed wrong reference 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 94275db..4c8d0e9 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -211,7 +211,7 @@ def _train_enc_dec_step_simple(encoder: model.Encoder, decoder: model.Decoder, reconstruction_loss = tf.losses.log_loss(inputs, x_decoded) - enc_dec_grads = tape.gradient(_enc_dec_train_loss, + enc_dec_grads = tape.gradient(reconstruction_loss, encoder.trainable_variables + decoder.trainable_variables) if int(global_step % LOG_FREQUENCY) == 0: summary_ops_v2.scalar(name='reconstruction_loss', tensor=reconstruction_loss,