From de6e15339fa1973e1c4621217d30b8c7746bc71d Mon Sep 17 00:00:00 2001 From: chenfeiyu Date: Fri, 17 Jul 2020 17:56:23 +0800 Subject: [PATCH] remove weigth norm when synthesizing to make inference fatser --- examples/deepvoice3/synthesize.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/examples/deepvoice3/synthesize.py b/examples/deepvoice3/synthesize.py index 39089b2..1f311fe 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()