diff --git a/src/transformers/utils/notebook.py b/src/transformers/utils/notebook.py index 9912b736b394..91e85a5d7ac3 100644 --- a/src/transformers/utils/notebook.py +++ b/src/transformers/utils/notebook.py @@ -33,15 +33,6 @@ def html_progress_bar(value, total, prefix, label, width=300): # docstyle-ignore return f"""
- {prefix} {label} @@ -327,6 +318,8 @@ def on_evaluate(self, args, state, control, metrics=None, **kwargs): values["Validation Loss"] = metrics["eval_loss"] _ = metrics.pop("total_flos", None) _ = metrics.pop("epoch", None) + _ = metrics.pop("eval_runtime", None) + _ = metrics.pop("eval_samples_per_second", None) for k, v in metrics.items(): if k == "eval_loss": values["Validation Loss"] = v