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

Fixed save location for models and the weights

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-19 11:24:42 +02:00
parent c6123607d9
commit d1ee766579
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/twomartens/masterthesis/ssd.py
View File

@ -479,12 +479,12 @@ def train_keras(train_generator: callable,
]
)
checkpoint_dir = os.path.join(weights_prefix, str(iteration), "/")
checkpoint_dir = os.path.join(weights_prefix, str(iteration))
os.makedirs(checkpoint_dir, exist_ok=True)
callbacks = [
tf.keras.callbacks.ModelCheckpoint(
filepath=f"{checkpoint_dir}ssd300-{{epoch:02d}}_loss-{{loss:.4f}}_val_loss-{{val_loss:.4f}}.h5",
filepath=f"{checkpoint_dir}/ssd300-{{epoch:02d}}_loss-{{loss:.4f}}_val_loss-{{val_loss:.4f}}.h5",
monitor="val_loss",
verbose=1,
save_best_only=True,