@@ -667,8 +667,10 @@ def get_common_parser(parser):
667
667
668
668
669
669
# 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 ,
672
674
help = "set IDs of GPUs to use" )
673
675
674
676
# profiling flags
@@ -679,26 +681,26 @@ def get_common_parser(parser):
679
681
# cyclic learning rate
680
682
parser .add_argument ("--clr_flag" ,
681
683
default = argparse .SUPPRESS ,
682
- #default=None,
684
+ #default=None,
683
685
type = str2bool ,
684
- help = "CLR flag (boolean)" )
686
+ help = "CLR flag (boolean)" )
685
687
parser .add_argument ("--clr_mode" ,
686
688
default = argparse .SUPPRESS ,
687
- #default=None,
689
+ #default=None,
688
690
type = str , choices = ['trng1' , 'trng2' , 'exp' ],
689
- help = "CLR mode (default: trng1)" )
691
+ help = "CLR mode (default: trng1)" )
690
692
parser .add_argument ("--clr_base_lr" , type = float ,
691
693
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." )
694
696
parser .add_argument ("--clr_max_lr" , type = float ,
695
697
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." )
698
700
parser .add_argument ("--clr_gamma" , type = float ,
699
701
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." )
702
704
703
705
return parser
704
706
@@ -949,7 +951,6 @@ def read_config_file(self, file):
949
951
for k ,v in config .items (sec ):
950
952
if not k in fileParams :
951
953
fileParams [k ] = eval (v )
952
-
953
954
fileParams = self .format_benchmark_config_arguments (fileParams )
954
955
#pprint(fileParams)
955
956
0 commit comments