From ab52f0feb0d58f23c624cd301b9675b9e88c884e Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 8 Feb 2019 08:30:06 +0100 Subject: [PATCH] Improved names of loss metrics Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index 9f546e6..64b3da1 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -120,9 +120,9 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i # define loss variables encoder_loss_avg = tfe.metrics.Mean(name='encoder_loss', dtype=tf.float32) decoder_loss_avg = tfe.metrics.Mean(name='decoder_loss', dtype=tf.float32) - enc_dec_loss_avg = tfe.metrics.Mean(name='enc_dec_loss', dtype=tf.float32) - zd_loss_avg = tfe.metrics.Mean(name='zd_loss', dtype=tf.float32) - xd_loss_avg = tfe.metrics.Mean(name='xd_loss', dtype=tf.float32) + enc_dec_loss_avg = tfe.metrics.Mean(name='encoder_decoder_loss', dtype=tf.float32) + zd_loss_avg = tfe.metrics.Mean(name='z_discriminator_loss', dtype=tf.float32) + xd_loss_avg = tfe.metrics.Mean(name='x_discriminator_loss', dtype=tf.float32) epoch_start_time = time.time()