From 74bd0f0cdce350e9ca30b937fb5fc7b3d17242fb Mon Sep 17 00:00:00 2001 From: mrq Date: Thu, 11 May 2023 03:30:54 +0000 Subject: [PATCH] revert local change that made its way upstream (showing graphs by it instead of epoch) --- src/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/utils.py b/src/utils.py index 597f547..2b3b52e 100755 --- a/src/utils.py +++ b/src/utils.py @@ -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 )