From 8d85d01c8a5891f71684c301fd601417de794cf7 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 25 Apr 2018 05:36:33 -0700 Subject: [PATCH] bug fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 4108f115..2927dd88 100644 --- a/train.py +++ b/train.py @@ -20,7 +20,7 @@ from utils.generic_utils import (Progbar, remove_experiment_folder, create_experiment_folder, save_checkpoint, save_best_model, load_config, lr_decay, count_parameters, check_update, get_commit_hash, - create_attn_mask) + create_attn_mask, mk_decay) from utils.model import get_param_size from utils.visual import plot_alignment, plot_spectrogram from models.tacotron import Tacotron