From 235ce071c6a419658cb48d4c8aa21d210b9e39ae Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Thu, 1 Feb 2018 08:43:59 -0800 Subject: [PATCH] Lr update --- datasets/.LJSpeech.py.swp | Bin 12288 -> 12288 bytes datasets/LJSpeech.py | 2 ++ train.py | 4 ++-- 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/datasets/.LJSpeech.py.swp b/datasets/.LJSpeech.py.swp index 2cc0c4af2ea4ca3a898e1cbff8b62ccef80cf44d..72acb3f71f5bdb30fde967ffeba971bccb3de835 100644 GIT binary patch delta 270 zcmZojXh;xCG6?hZRWR2xVE_UF28LqC;;6UE8^wO{GX`uH6!^@~+seei@Q@iI8!%Z= z;eOg}b_RykKwJ;RbwKP0#9To9jE#ZeDG<*E;s_u%24W*1e#^?hunmZ_fmjEKwSkx) zh;OklFl+?k4M5xh#O*+A4#Xlr{E&I$LOZ6(&MH2W1+@5pm}hg0%3kJ>+{B9blGKWl z_?*-{1zUx@0zIIJ223QqB%@f9K>-QCWim>Nf$F1y+B7OO71HvH6e<)l^Atexo4GZ< GvI78IK0rAD delta 173 zcmZojXh;xCG6?hZRWR2xVE_UF28Qx08By<5H;Vn@XY|`FDDat|_dOE>gEI?6)^D<) z!u_O&>5W55Me>Mh&T|hh)h^GLtKM+d*@mW>|hCU$f1>!;=wgF;c qAQl4RS3r|r0`W{BE(2naO-?Ku7k*@#JVn`Ov#v@P^JYnn=j;HHK_h$s diff --git a/datasets/LJSpeech.py b/datasets/LJSpeech.py index 4c4ddc2d..b15bc29e 100644 --- a/datasets/LJSpeech.py +++ b/datasets/LJSpeech.py @@ -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 diff --git a/train.py b/train.py index 72d58d19..3bf3f4ee 100644 --- a/train.py +++ b/train.py @@ -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