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

Improved paths for saved model weights

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-02-08 09:07:33 +01:00
parent cc5e7f89a2
commit 9d0f93689e
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

@ -219,10 +219,10 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i
print("Training finish!... save training results") print("Training finish!... save training results")
# save trained models # save trained models
encoder.save_weights("./weights/encoder") encoder.save_weights("./weights/encoder/")
decoder.save_weights("./weights/decoder") decoder.save_weights("./weights/decoder/")
z_discriminator.save_weights("./weights/z_discriminator") z_discriminator.save_weights("./weights/z_discriminator/")
x_discriminator.save_weights("./weights/x_discriminator") x_discriminator.save_weights("./weights/x_discriminator/")
def train_xdiscriminator_step(x_discriminator: XDiscriminator, decoder: Decoder, def train_xdiscriminator_step(x_discriminator: XDiscriminator, decoder: Decoder,