From 3f08c546797b1631dcaa415a37438b75e314453c Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 8 Feb 2019 10:31:51 +0100 Subject: [PATCH] Turned-off implicit summary saving at end of epoch Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index 45ca96e..0e0e6fe 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -203,10 +203,11 @@ 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_avg.result()}, X Discriminator loss: {xd_loss_avg.result():.3f}, " - f"Z Discriminator loss: {zd_loss_avg.result():.3f}, " - f"Encoder + Decoder loss: {enc_dec_loss_avg.result():.3f}, " - f"Encoder loss: {encoder_loss_avg.result():.3f}" + f"Decoder loss: {decoder_loss_avg.result(False)}, " + f"X Discriminator loss: {xd_loss_avg.result(False):.3f}, " + f"Z Discriminator loss: {zd_loss_avg.result(False):.3f}, " + f"Encoder + Decoder loss: {enc_dec_loss_avg.result(False):.3f}, " + f"Encoder loss: {encoder_loss_avg.result(False):.3f}" )) # save sample image