[Enhance] Use `--a-b` instead of `--a_b` in arguments. (#754)
parent
292127098f
commit
a87ff7e153
|
@ -104,7 +104,7 @@ def main():
|
|||
parser.add_argument(
|
||||
'--bn', action='store_true', help='whether original vgg has BN')
|
||||
parser.add_argument(
|
||||
'--layer_num',
|
||||
'--layer-num',
|
||||
type=int,
|
||||
choices=[11, 13, 16, 19],
|
||||
default=11,
|
||||
|
|
|
@ -97,7 +97,7 @@ def parse_args():
|
|||
choices=['none', 'pytorch', 'slurm', 'mpi'],
|
||||
default='none',
|
||||
help='job launcher')
|
||||
parser.add_argument('--local_rank', type=int, default=0)
|
||||
parser.add_argument('--local-rank', type=int, default=0)
|
||||
args = parser.parse_args()
|
||||
if 'LOCAL_RANK' not in os.environ:
|
||||
os.environ['LOCAL_RANK'] = str(args.local_rank)
|
||||
|
|
|
@ -93,7 +93,7 @@ def parse_args():
|
|||
choices=['none', 'pytorch', 'slurm', 'mpi'],
|
||||
default='none',
|
||||
help='job launcher')
|
||||
parser.add_argument('--local_rank', type=int, default=0)
|
||||
parser.add_argument('--local-rank', type=int, default=0)
|
||||
args = parser.parse_args()
|
||||
if 'LOCAL_RANK' not in os.environ:
|
||||
os.environ['LOCAL_RANK'] = str(args.local_rank)
|
||||
|
|
|
@ -73,7 +73,7 @@ def parse_args():
|
|||
choices=['none', 'pytorch', 'slurm', 'mpi'],
|
||||
default='none',
|
||||
help='job launcher')
|
||||
parser.add_argument('--local_rank', type=int, default=0)
|
||||
parser.add_argument('--local-rank', type=int, default=0)
|
||||
args = parser.parse_args()
|
||||
if 'LOCAL_RANK' not in os.environ:
|
||||
os.environ['LOCAL_RANK'] = str(args.local_rank)
|
||||
|
|
Loading…
Reference in New Issue