diff --git a/examples/deepvoice3/synthesize.py b/examples/deepvoice3/synthesize.py index 39089b2a58ca19a06b3f31fc74baeb2c60029090..1f311fef5b4d10744ec8f14a63fdf7c0e741fefa 100644 --- a/examples/deepvoice3/synthesize.py +++ b/examples/deepvoice3/synthesize.py @@ -17,7 +17,7 @@ import soundfile as sf from parakeet.data import SliceDataset, DataCargo, PartialyRandomizedSimilarTimeLengthSampler, SequentialSampler from parakeet.utils.io import save_parameters, load_parameters, add_yaml_config_to_args from parakeet.g2p import en - +from parakeet.models.deepvoice3.weight_norm_hook import remove_weight_norm from vocoder import WaveflowVocoder, GriffinLimVocoder from train import create_model @@ -25,6 +25,12 @@ from train import create_model def main(args, config): model = create_model(config) loaded_step = load_parameters(model, checkpoint_path=args.checkpoint) + for name, layer in model.named_sublayers(): + try: + remove_weight_norm(layer) + except ValueError: + # this layer has not weight norm hook + pass model.eval() if args.vocoder == "waveflow": vocoder = WaveflowVocoder()