From eb1a08f7d4cfda9403ec6d1a036418cb2f8b524c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= Date: Tue, 5 Sep 2023 20:15:42 +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 b31a8bf..3f10a75 100644 --- a/bert_gen.py +++ b/bert_gen.py @@ -42,7 +42,7 @@ def process_line(line): if __name__ == "__main__": - parser.add_argument('-c', '--config', type=str, default=None) + parser.add_argument('-c', '--config', type=str, default="configs/config.json") config_path = args.config hps = utils.get_hparams_from_file(config_path) lines = []