Commit 1bc98a63 authored by sjjsmuel's avatar sjjsmuel

save only the best Model

parent 268c6a29
......@@ -121,7 +121,7 @@ model.compile(optimizer=RMSprop())
callbacks = [
ModelCheckpoint(
filepath= str(checkpoint_path) + '/model.{epoch:03d}-{val_loss:.3f}.hdf5',
filepath= str(checkpoint_path) + '/best_model.hdf5',
# Path where to save the model
# The two parameters below mean that we will overwrite
# the current checkpoint if and only if
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment