[Fix bug] remove n processes of embed in pre.py

3 years ago · 99269b2046
parent 28e6bce570
commit 99269b2046
1 changed files with 3 additions and 2 deletions
--- 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)