malarinv/tacotron2
1
0
Fork 0
mirror of https://github.com/malarinv/tacotron2 synced 2026-03-08 01:32:35 +00:00
Code Issues Releases Wiki Activity

hparams.py: adding use saved learning rate param

Browse Source
This commit is contained in:
rafaelvalle
2018-06-05 08:12:49 -07:00
parent 22bcff1155
commit 5f0ea06c41
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
train.py
View File

@@ -190,8 +190,11 @@ def train(output_directory, log_directory, checkpoint_path, warm_start, n_gpus,
if warm_start:
model = warm_start_model(checkpoint_path, model)
else:
model, optimizer, learning_rate, iteration = load_checkpoint(
model, optimizer, _learning_rate, iteration = load_checkpoint(
checkpoint_path, model, optimizer)
if hparams.use_saved_learning_rate:
learning_rate = _learning_rate
iteration += 1 # next iteration is iteration + 1
epoch_offset = max(0, int(iteration / len(train_loader)))