revert local change that made its way upstream (showing graphs by it instead of epoch)

This commit is contained in:
mrq 2023-05-11 03:30:54 +00:00
parent 149aaca554
commit 74bd0f0cdc

View File

@ -1882,7 +1882,7 @@ def update_training_dataplot(x_min=None, x_max=None, y_min=None, y_max=None, con
losses = gr.LinePlot.update(
value = pd.DataFrame(training_state.statistics['loss']),
x_lim=x_lim, y_lim=y_lim,
x="it", y="value", # x="epoch",
x="epoch", y="value", # x="it",
title="Loss Metrics", color="type", tooltip=['epoch', 'it', 'value', 'type'],
width=500, height=350
)
@ -1890,7 +1890,7 @@ def update_training_dataplot(x_min=None, x_max=None, y_min=None, y_max=None, con
lrs = gr.LinePlot.update(
value = pd.DataFrame(training_state.statistics['lr']),
x_lim=x_lim,
x="it", y="value", # x="epoch",
x="epoch", y="value", # x="it",
title="Learning Rate", color="type", tooltip=['epoch', 'it', 'value', 'type'],
width=500, height=350
)
@ -1898,7 +1898,7 @@ def update_training_dataplot(x_min=None, x_max=None, y_min=None, y_max=None, con
grad_norms = gr.LinePlot.update(
value = pd.DataFrame(training_state.statistics['grad_norm']),
x_lim=x_lim,
x="it", y="value", # x="epoch",
x="epoch", y="value", # x="it",
title="Gradient Normals", color="type", tooltip=['epoch', 'it', 'value', 'type'],
width=500, height=350
)