From f945aa6091dc59241972390494cecff6bdae22b3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Tue, 5 Sep 2023 19:31:35 +0800 Subject: [PATCH] Update bert_gen.py --- bert_gen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bert_gen.py b/bert_gen.py index 2259454..f71ba77 100644 --- a/bert_gen.py +++ b/bert_gen.py @@ -46,7 +46,7 @@ if __name__ == "__main__": with open(hps.data.validation_files, encoding="utf-8") as f: lines.extend(f.readlines()) - # A100 40GB suitable config, if coom, please decrease the processes number. + # RTX4090 24GB suitable config, if coom, please decrease the processes number. with Pool(processes=6) as pool: for _ in tqdm(pool.imap_unordered(process_line, lines), total=len(lines)): pass