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

Improved formatting

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-11 12:22:10 +02:00
parent 0ae33130ad
commit ba2bd776ea
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -312,12 +312,12 @@ def train(train_generator: callable,
callbacks.append(tensorboard_callback)
history = ssd_model.fit_generator(generator=train_generator,
epochs=nr_epochs,
steps_per_epoch=steps_per_epoch_train,
validation_data=val_generator,
validation_steps=steps_per_epoch_val,
callbacks=callbacks,
initial_epoch=initial_epoch)
epochs=nr_epochs,
steps_per_epoch=steps_per_epoch_train,
validation_data=val_generator,
validation_steps=steps_per_epoch_val,
callbacks=callbacks,
initial_epoch=initial_epoch)
ssd_model.save(f"{checkpoint_dir}/ssd300.h5")
ssd_model.save_weights(f"{checkpoint_dir}/ssd300_weights.h5")