Merge pull request Rayhane-mamah#133 from hiroto911/fix_load_model

Fix bug loading model
gloriouskilka · Aug 8, 2018 · a94b9d2 · a94b9d2
2 parents e2c2275 + 15dcb4d
commit a94b9d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tacotron/train.py b/tacotron/train.py
@@ -157,7 +157,7 @@ def train(log_dir, args, hparams):
 
 					if (checkpoint_state and checkpoint_state.model_checkpoint_path):
 						log('Loading checkpoint {}'.format(checkpoint_state.model_checkpoint_path), slack=True)
-						load_averaged_model(sess, sh_saver, checkpoint_state.model_checkpoint_path)
+						saver.restore(sess, checkpoint_state.model_checkpoint_path)
 					else:
 						log('No model to load at {}'.format(save_dir), slack=True)