Skip to content

GitLab

Unverified Commit a64df04b authored by Leo Gao's avatar Leo Gao Committed by GitHub
Browse files

Update main.py

parent 7d5aa3f7
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
main.py
View file @ a64df04b
...@@ -30,6 +30,9 @@ def main(): ...@@ -30,6 +30,9 @@ def main():
np.random.seed(args.seed) np.random.seed(args.seed)
lm = models.get_model(args.model).create_from_arg_string(args.model_args) lm = models.get_model(args.model).create_from_arg_string(args.model_args)
if args.limit:
print("WARNING: --limit SHOULD ONLY BE USED FOR TESTING. REAL METRICS SHOULD NOT BE COMPUTED USING LIMIT.")
if not args.no_cache: if not args.no_cache:
lm = base.CachingLM(lm, 'lm_cache/' + args.model + '_' + args.model_args.replace('=', '-').replace(',', '_') + '.db') lm = base.CachingLM(lm, 'lm_cache/' + args.model + '_' + args.model_args.replace('=', '-').replace(',', '_') + '.db')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment