Merge branch 'dev' of https://github.com/mozilla/TTS into dev

2021-01-22 13:57:57 +01:00 · 2021-01-22 13:57:57 +01:00 · 503ae3ee76
parent 1d3de15b16 fae10309e4
commit 503ae3ee76
1 changed files with 1 additions and 1 deletions
--- a/TTS/bin/train_tacotron.py
+++ b/TTS/bin/train_tacotron.py
@ -549,7 +549,7 @@ def main(args):  # pylint: disable=redefined-outer-name
                scaler.load_state_dict(checkpoint["scaler"])
            if c.reinit_layers:
                raise RuntimeError
-        except KeyError:
+        except (KeyError, RuntimeError):
            print(" > Partial model initialization.")
            model_dict = model.state_dict()
            model_dict = set_init_dict(model_dict, checkpoint['model'], c)