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

Lr update

Browse Source
This commit is contained in:
Eren Golge
2018-02-01 08:43:59 -08:00
parent 7304978be3
commit 235ce071c6
3 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
datasets/.LJSpeech.py.swp
BIN
View File
Binary file not shown.
datasets/LJSpeech.py
+2
View File
@@ -52,6 +52,8 @@ class LJSpeechDataset(Dataset):
keys = list()
text = [d['text'] for d in batch]
text_lenghts = [len(x) for x in text]
max_text_len = np.max(text_lengths)
wav = [d['wav'] for d in batch]
# PAD sequences with largest length of the batch
train.py
+2 -2
View File
@@ -97,7 +97,7 @@ def main(args):
else:
criterion = nn.L1Loss()
n_priority_freq = int(3000 / (c.sample_rate * 0.5) * c.num_freq)
n_priority_freq = int(3000 / (c.sample_rate * 0.5) * c.num_freq)
#lr_scheduler = ReduceLROnPlateau(optimizer, factor=c.lr_decay,
# patience=c.lr_patience, verbose=True)
@@ -121,7 +121,7 @@ n_priority_freq = int(3000 / (c.sample_rate * 0.5) * c.num_freq)
current_step = i + args.restore_step + epoch * len(dataloader) + 1
# setup lr
current_lr = lr_decay(init_lr, current_step)
current_lr = lr_decay(c.lr, current_step)
for params_group in optimizer.param_groups:
param_group['lr'] = current_lr