wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-11 12:00:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

plot attention alignments

Browse Source
This commit is contained in:
Eren Golge
2018-02-02 05:37:09 -08:00
parent 235ce071c6
commit 4e0ab65bbf
5 changed files with 66 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files
train.py
+5 -2
View File
@@ -123,7 +123,7 @@ def main(args):
# setup lr
current_lr = lr_decay(c.lr, current_step)
for params_group in optimizer.param_groups:
param_group['lr'] = current_lr
params_group['lr'] = current_lr
optimizer.zero_grad()
@@ -204,11 +204,14 @@ def main(args):
checkpoint_path)
print("\n | > Checkpoint is saved : {}".format(checkpoint_path))
# Log spectrogram reconstruction
# Diagnostic visualizations
const_spec = linear_output[0].data.cpu()[None, :]
gt_spec = linear_spec_var[0].data.cpu()[None, :]
align_img = alignments[0].data.cpu().t()[None, :]
tb.add_image('Spec/Reconstruction', const_spec, current_step)
tb.add_image('Spec/GroundTruth', gt_spec, current_step)
tb.add_image('Attn/Alignment', align_img, current_step)
#lr_scheduler.step(loss.data[0])
tb.add_scalar('Time/EpochTime', epoch_time, epoch)