Skip to content

GitLab

Commit 84f14daf authored by Leo Gao's avatar Leo Gao
Browse files

Cache by default

parent aab91285
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
main.py
View file @ 84f14daf
...@@ -20,7 +20,7 @@ def parse_args(): ...@@ -20,7 +20,7 @@ def parse_args():
parser.add_argument('--seed', type=int, default=1234) parser.add_argument('--seed', type=int, default=1234)
parser.add_argument('--output_path', default=None) parser.add_argument('--output_path', default=None)
parser.add_argument('--limit', type=int, default=None) parser.add_argument('--limit', type=int, default=None)
parser.add_argument('--cache', action="store_true") parser.add_argument('--no_cache', action="store_true")
return parser.parse_args() return parser.parse_args()
def main(): def main():
...@@ -31,7 +31,7 @@ def main(): ...@@ -31,7 +31,7 @@ def main():
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.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')
if args.tasks == "all_tasks": if args.tasks == "all_tasks":
task_names = tasks.ALL_TASKS task_names = tasks.ALL_TASKS
......
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