Skip to content

Commit 995b94f

Browse files
committed
Fixed 'gpus' issue, fixed formatting.
1 parent 752a47d commit 995b94f

File tree

1 file changed

+14
-13
lines changed

1 file changed

+14
-13
lines changed

common/default_utils.py

Lines changed: 14 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -667,8 +667,10 @@ def get_common_parser(parser):
667667

668668

669669
# Backend configuration
670-
parser.add_argument("--gpus", action="store", nargs='*',
671-
default=[], type=int,
670+
parser.add_argument("--gpus", nargs="*",
671+
default=argparse.SUPPRESS,
672+
#default=[0],
673+
type=int,
672674
help="set IDs of GPUs to use")
673675

674676
# profiling flags
@@ -679,26 +681,26 @@ def get_common_parser(parser):
679681
# cyclic learning rate
680682
parser.add_argument("--clr_flag",
681683
default=argparse.SUPPRESS,
682-
#default=None,
684+
#default=None,
683685
type=str2bool,
684-
help="CLR flag (boolean)")
686+
help="CLR flag (boolean)")
685687
parser.add_argument("--clr_mode",
686688
default=argparse.SUPPRESS,
687-
#default=None,
689+
#default=None,
688690
type=str, choices=['trng1', 'trng2', 'exp'],
689-
help="CLR mode (default: trng1)")
691+
help="CLR mode (default: trng1)")
690692
parser.add_argument("--clr_base_lr", type=float,
691693
default=argparse.SUPPRESS,
692-
#default=1e-4,
693-
help="Base lr for cycle lr.")
694+
#default=1e-4,
695+
help="Base lr for cycle lr.")
694696
parser.add_argument("--clr_max_lr", type=float,
695697
default=argparse.SUPPRESS,
696-
#default=1e-3,
697-
help="Max lr for cycle lr.")
698+
#default=1e-3,
699+
help="Max lr for cycle lr.")
698700
parser.add_argument("--clr_gamma", type=float,
699701
default=argparse.SUPPRESS,
700-
#default=0.999994,
701-
help="Gamma parameter for learning cycle LR.")
702+
#default=0.999994,
703+
help="Gamma parameter for learning cycle LR.")
702704

703705
return parser
704706

@@ -949,7 +951,6 @@ def read_config_file(self, file):
949951
for k,v in config.items(sec):
950952
if not k in fileParams:
951953
fileParams[k] = eval(v)
952-
953954
fileParams = self.format_benchmark_config_arguments(fileParams)
954955
#pprint(fileParams)
955956

0 commit comments

Comments
 (0)