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

shangdibufashi · Sep 24, 2021 · 99269b2 · 99269b2
1 parent 28e6bce
commit 99269b2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pre.py b/pre.py
@@ -68,7 +68,8 @@
     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)