graykode / commit-autosuggestions

Go to a project
Toggle navigation Toggle navigation pinning
graykode
705ee9a5 1 parent 9934e09b

(refactor) rename folder to support other language

Inline Side-by-side
Showing 4 changed files with 1 additions and 1 deletions
...@@ -146,7 +146,7 @@ def main(args): ...@@ -146,7 +146,7 @@ def main(args):
146 146
147 if __name__ == '__main__': 147 if __name__ == '__main__':
148 parser = argparse.ArgumentParser(description="") 148 parser = argparse.ArgumentParser(description="")
149 - parser.add_argument("--load_model_path", default='weight', type=str, 149 + parser.add_argument("--load_model_path", type=str, required=True,
150 help="Path to trained model: Should contain the .bin files") 150 help="Path to trained model: Should contain the .bin files")
151 151
152 parser.add_argument("--model_type", default='roberta', type=str, 152 parser.add_argument("--model_type", default='roberta', type=str,
......