From 948ef644233e8db3d74ab25b8cdc67225c5b3b9c Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 8 Feb 2019 13:41:09 +0100 Subject: [PATCH] Added formatting of decoder loss 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 998c10a..d7acd39 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -227,7 +227,7 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i print(( f"[{epoch + 1:d}/{train_epoch:d}] - " f"train time: {per_epoch_time:.2f}, " - f"Decoder loss: {decoder_loss}, " + f"Decoder loss: {decoder_loss:.3f}, " f"X Discriminator loss: {xd_loss:.3f}, " f"Z Discriminator loss: {zd_loss:.3f}, " f"Encoder + Decoder loss: {enc_dec_loss:.3f}, "