diff --git a/TTS/trainer.py b/TTS/trainer.py index 2175875c..e8911ba3 100644 --- a/TTS/trainer.py +++ b/TTS/trainer.py @@ -453,7 +453,7 @@ class Trainer: if "scaler" in checkpoint and self.use_amp_scaler and checkpoint["scaler"]: print(" > Restoring Scaler...") scaler = _restore_list_objs(checkpoint["scaler"], scaler) - except (KeyError, RuntimeError): + except (KeyError, RuntimeError, ValueError): print(" > Partial model initialization...") model_dict = model.state_dict() model_dict = set_init_dict(model_dict, checkpoint["model"], config)