bug fix for TTS model restore

2020-06-06 12:59:59 +02:00 · 2020-06-06 12:59:59 +02:00 · b86af6538f
parent 79db7d931b
commit b86af6538f
1 changed files with 1 additions and 1 deletions
--- 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: