From b736c6cdacf5e41ec78fb9c847e91975aebfa26c Mon Sep 17 00:00:00 2001 From: erogol Date: Sat, 6 Jun 2020 12:59:59 +0200 Subject: [PATCH] bug fix for TTS model restore --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 13bda5ef..a47828e0 100644 --- a/train.py +++ b/train.py @@ -542,7 +542,7 @@ def main(args): # pylint: disable=redefined-outer-name except: print(" > Partial model initialization.") model_dict = model.state_dict() - model_dict = set_init_dict(model_dict, checkpoint, c) + model_dict = set_init_dict(model_dict, checkpoint['model'], c) model.load_state_dict(model_dict) del model_dict for group in optimizer.param_groups: