diff --git a/pre.py b/pre.py index fbec51a..1d78a12 100644 --- a/pre.py +++ b/pre.py @@ -68,7 +68,8 @@ if __name__ == "__main__": del args.no_trim, args.encoder_model_fpath args.hparams = hparams.parse(args.hparams) - + n_processes_embed = args.n_processes_embed + del args.n_processes_embed preprocess_dataset(**vars(args)) - create_embeddings(synthesizer_root=args.out_dir, n_processes=args.n_processes_embed, encoder_model_fpath=encoder_model_fpath) + create_embeddings(synthesizer_root=args.out_dir, n_processes=n_processes_embed, encoder_model_fpath=encoder_model_fpath)