Update bert_gen.py

2023-09-05 20:17:03 +08:00
parent eb1a08f7d4
commit a716d8ef82
1 changed files with 1 additions and 0 deletions
--- a/bert_gen.py
+++ b/bert_gen.py
@@ -42,6 +42,7 @@ def process_line(line):


 if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
    parser.add_argument('-c', '--config', type=str, default="configs/config.json")
    config_path = args.config
    hps = utils.get_hparams_from_file(config_path)