Skip to content

Commit

Permalink
Unifying the format of argparse (facebookresearch#24)
Browse files Browse the repository at this point in the history
* Unifying the format of argparse

* Minor fixes
  • Loading branch information
zhiqwang authored May 30, 2020
1 parent 0af4193 commit 0fb754c
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
1 change: 1 addition & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -14,3 +14,4 @@ experiment-*
**/.mypy_cache/*
.mypy_cache/*
not_tracked_dir/
.vscode
8 changes: 4 additions & 4 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down

0 comments on commit 0fb754c

Please sign in to comment.