From 862fc04042eebf688bf8fff8a5fd114dcfd44aa7 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <2225664821@qq.com>
Date: Thu, 3 Aug 2023 18:37:06 +0800
Subject: [PATCH] Update spec_gen.py

---
 spec_gen.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spec_gen.py b/spec_gen.py
index 3aee5e3..80c18c8 100644
--- 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