2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed wrong reference

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-15 12:16:38 +02:00
parent 031a1f9ef0
commit 2e076597ef
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/aae/train.py
View File

@ -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,