From 85d743c058b9e344191e298c7efa68df07403fab Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 8 Feb 2019 10:46:25 +0100 Subject: [PATCH] Added inlier class to saved weights Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index 0d0f244..504bf12 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -220,10 +220,10 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i print("Training finish!... save training results") # save trained models - encoder.save_weights("./weights/encoder/") - decoder.save_weights("./weights/decoder/") - z_discriminator.save_weights("./weights/z_discriminator/") - x_discriminator.save_weights("./weights/x_discriminator/") + encoder.save_weights("./weights/encoder/" + str(inlier_classes[0])) + decoder.save_weights("./weights/decoder/" + str(inlier_classes[0])) + z_discriminator.save_weights("./weights/z_discriminator/" + str(inlier_classes[0])) + x_discriminator.save_weights("./weights/x_discriminator/" + str(inlier_classes[0])) def train_xdiscriminator_step(x_discriminator: XDiscriminator, decoder: Decoder,