diff options
| author | TharinduDR <rhtdranasinghe@gmail.com> | 2020-07-09 14:30:05 +0300 |
|---|---|---|
| committer | TharinduDR <rhtdranasinghe@gmail.com> | 2020-07-09 14:30:05 +0300 |
| commit | 35cc34dfd6b9ad373f37af9731d06309602d21cb (patch) | |
| tree | 0d4bbf6e68f4ad36a69a797965f0e1bff9742128 | |
| parent | b7023fa2ab47bd4f5f806623b1aa730f3b11edf4 (diff) | |
031: Fine tunning
| -rw-r--r-- | transquest/algo/transformers/run_model.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/transquest/algo/transformers/run_model.py b/transquest/algo/transformers/run_model.py index 2af31bc..75dbe15 100644 --- a/transquest/algo/transformers/run_model.py +++ b/transquest/algo/transformers/run_model.py @@ -438,11 +438,12 @@ class QuestModel: early_stopping_counter = 0 else: if args["use_early_stopping"]: + loss_name = args["early_stopping_metric"] if early_stopping_counter < args["early_stopping_patience"]: early_stopping_counter += 1 if verbose: print() - print(f"No improvement in eval_loss for {early_stopping_counter} steps.") + print(f"No improvement in {loss_name} for {early_stopping_counter} steps.") print(f"Training will stop at {args['early_stopping_patience']} steps.") print() else: |