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

Fixed save dir for models

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-07 13:39:45 +02:00
parent e3a05eb5fc
commit 322f2c51ab
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

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