mrq/ai-voice-cloning
31
74
Fork 43
Code Issues 223 Pull Requests 5 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
mrq 2023-05-11 03:30:54 +00:00
parent 149aaca554
commit 74bd0f0cdc
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/utils.py
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( losses = gr.LinePlot.update(
value = pd.DataFrame(training_state.statistics['loss']), value = pd.DataFrame(training_state.statistics['loss']),
x_lim=x_lim, y_lim=y_lim, 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'], title="Loss Metrics", color="type", tooltip=['epoch', 'it', 'value', 'type'],
width=500, height=350 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( lrs = gr.LinePlot.update(
value = pd.DataFrame(training_state.statistics['lr']), value = pd.DataFrame(training_state.statistics['lr']),
x_lim=x_lim, 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'], title="Learning Rate", color="type", tooltip=['epoch', 'it', 'value', 'type'],
width=500, height=350 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( grad_norms = gr.LinePlot.update(
value = pd.DataFrame(training_state.statistics['grad_norm']), value = pd.DataFrame(training_state.statistics['grad_norm']),
x_lim=x_lim, 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'], title="Gradient Normals", color="type", tooltip=['epoch', 'it', 'value', 'type'],
width=500, height=350 width=500, height=350
) )