From 4c25d8aaf7aa4e987a89231818297c934e57ae68 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <star_dust_chen@foxmail.com>
Date: Tue, 5 Sep 2023 22:34:27 +0800
Subject: [PATCH] Update bert_gen.py

---
 bert_gen.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/bert_gen.py b/bert_gen.py
index f13ff4d..99a7ee3 100644
--- a/bert_gen.py
+++ b/bert_gen.py
@@ -43,9 +43,7 @@ def process_line(line):
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument('-c', '--config', type=str, default="configs/config.json")
-    parser.add_argument(
-        '--num_processes', type=int, default=1, help='You are advised to set the number of processes to the same as the number of CPU cores'
-    )
+    parser.add_argument('--num_processes', type=int, default=2')
     args = parser.parse_args()
     config_path = args.config
     hps = utils.get_hparams_from_file(config_path)