Unverified Commit 02f7c2fe authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Some styling of the training table in Notebooks (#11118)

parent 11505fa1
......@@ -33,15 +33,6 @@ def html_progress_bar(value, total, prefix, label, width=300):
# docstyle-ignore
return f"""
<div>
<style>
/* Turns off some styling */
progress {{
/* gets rid of default border in Firefox and Opera. */
border: none;
/* Needs to be in here for Safari polyfill so background images work as expected. */
background-size: auto;
}}
</style>
{prefix}
<progress value='{value}' max='{total}' style='width:{width}px; height:20px; vertical-align: middle;'></progress>
{label}
......@@ -327,6 +318,8 @@ class NotebookProgressCallback(TrainerCallback):
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
......
Markdown is supported
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