Mirrors/TTS
1
0
Fork 0
mirror of https://github.com/coqui-ai/TTS.git synced 2025-12-25 12:49:29 +01:00
Code Issues Packages Projects Releases Wiki Activity

log grad norm on tf

Browse Source
This commit is contained in:
Eren Golge
2018-02-01 07:39:35 -08:00
parent 088a105a43
commit 3519f2f33c
4 changed files with 85 additions and 367 deletions
Download Patch File Download Diff File Expand all files Collapse all files

285
PlayGround.ipynb
View File

File diff suppressed because one or more lines are too long

121
notebooks/PlayGround.ipynb
View File

File diff suppressed because one or more lines are too long

36
notebooks/TestModelOnLJSpeech.ipynb
View File

File diff suppressed because one or more lines are too long

10
train.py
View File

@@ -174,12 +174,14 @@ def main(args):
('mel_loss', mel_loss.data[0]),
('grad_norm', grad_norm)])
tb.add_scalar('Train/TotalLoss', loss.data[0], current_step)
tb.add_scalar('Train/LinearLoss', linear_loss.data[0],
tb.add_scalar('Loss/TotalLoss', loss.data[0], current_step)
tb.add_scalar('Loss/LinearLoss', linear_loss.data[0],
current_step)
tb.add_scalar('Train/MelLoss', mel_loss.data[0], current_step)
tb.add_scalar('LearningRate', optimizer.param_groups[0]['lr'],
tb.add_scalar('Loss/MelLoss', mel_loss.data[0], current_step)
tb.add_scalar('Params/LearningRate', optimizer.param_groups[0]['lr'],
current_step)
tb.add_scalar('Params/GradNorm', grad_norm, current_step)
tb.add_scalar('Time/StepTime', step_time, current_step)