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')