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): ...@@ -33,15 +33,6 @@ def html_progress_bar(value, total, prefix, label, width=300):
# docstyle-ignore # docstyle-ignore
return f""" return f"""
<div> <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} {prefix}
<progress value='{value}' max='{total}' style='width:{width}px; height:20px; vertical-align: middle;'></progress> <progress value='{value}' max='{total}' style='width:{width}px; height:20px; vertical-align: middle;'></progress>
{label} {label}
...@@ -327,6 +318,8 @@ class NotebookProgressCallback(TrainerCallback): ...@@ -327,6 +318,8 @@ class NotebookProgressCallback(TrainerCallback):
values["Validation Loss"] = metrics["eval_loss"] values["Validation Loss"] = metrics["eval_loss"]
_ = metrics.pop("total_flos", None) _ = metrics.pop("total_flos", None)
_ = metrics.pop("epoch", None) _ = metrics.pop("epoch", None)
_ = metrics.pop("eval_runtime", None)
_ = metrics.pop("eval_samples_per_second", None)
for k, v in metrics.items(): for k, v in metrics.items():
if k == "eval_loss": if k == "eval_loss":
values["Validation Loss"] = v 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