From 279e848434726755c692b171b1bb648ae363d835 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Fri, 8 Feb 2019 14:59:50 +0100 Subject: [PATCH] Fixed specification of checkpoint dir Signed-off-by: Jim Martens --- src/twomartens/masterthesis/aae/train.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/aae/train.py b/src/twomartens/masterthesis/aae/train.py index b288d10..2f7d700 100644 --- a/src/twomartens/masterthesis/aae/train.py +++ b/src/twomartens/masterthesis/aae/train.py @@ -127,8 +127,10 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i total_lowest_loss = math.inf grace_period = GRACE - checkpoint_prefix = './weights/' + str(inlier_classes[0]) + '/' + str(iteration) + '/ckpt' - latest_checkpoint = tf.train.latest_checkpoint(checkpoint_prefix) + checkpoint_dir = os.path.abspath('./weights/' + str(inlier_classes[0]) + '/' + str(iteration) + '/') + os.makedirs(checkpoint_dir, exist_ok=True) + checkpoint_prefix = os.path.join(checkpoint_dir, 'ckpt') + latest_checkpoint = tf.train.latest_checkpoint(checkpoint_dir) checkpoint = tf.train.Checkpoint(encoder=encoder, decoder=decoder, z_discriminator=z_discriminator,