Change default mode to "min" in best_checkpoint_callback

mila-iqia / cookiecutter-pyml

MIT License

19 stars 8 forks source link

Change default mode to "min" in best_checkpoint_callback #120

Closed ma-beaudoin closed 1 month ago

ma-beaudoin commented 1 month ago

By default, best_checkpoint_callback monitors val_loss with mode = "max", therefore the best models don't get saved as expected. Proposed fix: change the mode to "min".