From 0fb754c4e592b4da87e6ec0010c767265809666e Mon Sep 17 00:00:00 2001 From: Zhiqiang Wang Date: Sat, 30 May 2020 23:57:24 +0800 Subject: [PATCH] Unifying the format of argparse (#24) * Unifying the format of argparse * Minor fixes --- .gitignore | 1 + main.py | 8 ++++---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/.gitignore b/.gitignore index 34a68ade0..7d776a671 100644 --- a/.gitignore +++ b/.gitignore @@ -14,3 +14,4 @@ experiment-* **/.mypy_cache/* .mypy_cache/* not_tracked_dir/ +.vscode diff --git a/main.py b/main.py index e74466f49..e5f9eff80 100644 --- a/main.py +++ b/main.py @@ -84,21 +84,21 @@ def get_args_parser(): parser.add_argument('--coco_panoptic_path', type=str) parser.add_argument('--remove_difficult', action='store_true') - parser.add_argument('--output-dir', default='', + parser.add_argument('--output_dir', default='', help='path where to save, empty for no saving') parser.add_argument('--device', default='cuda', help='device to use for training / testing') parser.add_argument('--seed', default=42, type=int) parser.add_argument('--resume', default='', help='resume from checkpoint') - parser.add_argument('--start-epoch', default=0, type=int, metavar='N', + parser.add_argument('--start_epoch', default=0, type=int, metavar='N', help='start epoch') parser.add_argument('--eval', action='store_true') parser.add_argument('--num_workers', default=2, type=int) # distributed training parameters - parser.add_argument('--world-size', default=1, type=int, + parser.add_argument('--world_size', default=1, type=int, help='number of distributed processes') - parser.add_argument('--dist-url', default='env://', help='url used to set up distributed training') + parser.add_argument('--dist_url', default='env://', help='url used to set up distributed training') return parser