Commit 16f2d5ce authored by DGarbanzo's avatar DGarbanzo
Browse files

Fix train script

* Check point callback was initialised with on_best_model callback which overwrote the model on the end of each epoch, even when the valuation loss did not improve
* Fix passing on_best_model callback only if cfg.SEND_BEST_MODEL is set in the config
* Delete model.save() on_best_model as model saving is performed by Keras base class of MyCPCallback.
parent 98679f71
Supports Markdown
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