Update spec_gen.py

2023-08-03 18:37:06 +08:00
parent 5be73d35c0
commit 862fc04042
1 changed files with 1 additions and 1 deletions
--- a/spec_gen.py
+++ b/spec_gen.py
@@ -43,6 +43,6 @@ def process_line(line):
 with open(hps.data.training_files) as f:
    lines = f.readlines()

-with Pool() as pool:
+with Pool(processes=12) as pool: #A100 suitable config,if coom,please decrease the processess number.
    for _ in tqdm(pool.imap_unordered(process_line, lines)):
        pass