make metric configurable that triggers checkpoint writing #4
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Introducing a third configurable metric:
eval_metric
: validation metrics like BLEU or ChrF; measured on the validation set onlyschedule_metric
: decides when the scheduler decreases the learning rate or stops learning; might beloss
orppl
, might not matcheval_metric
(since often smoother curve)ckpt_metric
: decides when the model parameters are saved, i.e., a checkpoint is written; default iseval_metric
Before, checkpoint saving was coupled to
eval_metric
(new default), but potentially not matchingschedule_metric
(e.g., validating models with BLEU and storing when highest reached, but adapting learning rate according to loss). Now, one can decide whether all three align or are individually set.