From 4d15cb31b07f1647c3564a09738f2a2735fb7f85 Mon Sep 17 00:00:00 2001 From: Shawn-Shan Date: Mon, 13 Jul 2020 19:07:08 -0500 Subject: [PATCH] add local eval --- evaluation/eval_local.py | 1 + 1 file changed, 1 insertion(+) diff --git a/evaluation/eval_local.py b/evaluation/eval_local.py index fa5eb13..f828a3e 100644 --- a/evaluation/eval_local.py +++ b/evaluation/eval_local.py @@ -162,6 +162,7 @@ def main(): def parse_arguments(argv): parser = argparse.ArgumentParser() + parser.add_argument('--gpu', type=str, help='GPU id', default='0')