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

Added inlier class to saved weights

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-02-08 10:46:25 +01:00
parent ce6cda5351
commit 85d743c058
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

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