From 322f2c51ab706464030d5dc3e8f60d3c701eb2a1 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 7 Jul 2019 13:39:45 +0200 Subject: [PATCH] Fixed save dir for models Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index edf5601..07f5f23 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -504,8 +504,8 @@ def train_keras(train_generator: callable, callbacks=callbacks, initial_epoch=initial_epoch) - ssd_model.model.save(f"{checkpoint_dir}ssd300.h5") - ssd_model.model.save_weights(f"{checkpoint_dir}ssd300_weights.h5") + ssd_model.model.save(f"{checkpoint_dir}/ssd300.h5") + ssd_model.model.save_weights(f"{checkpoint_dir}/ssd300_weights.h5") return history