update CVPR-2019-GDAS re-train NASNet-search-space searched models
This commit is contained in:
@@ -12,6 +12,7 @@ def obtain_basic_args():
|
||||
parser.add_argument('--optim_config', type=str, help='The path to the optimizer configuration')
|
||||
parser.add_argument('--procedure' , type=str, help='The procedure basic prefix.')
|
||||
parser.add_argument('--model_source', type=str, default='normal',help='The source of model defination.')
|
||||
parser.add_argument('--extra_model_path', type=str, default=None, help='The extra model ckp file (help to indicate the searched architecture).')
|
||||
add_shared_args( parser )
|
||||
# Optimization options
|
||||
parser.add_argument('--batch_size', type=int, default=2, help='Batch size for training.')
|
||||
|
@@ -29,7 +29,8 @@ def convert_param(original_lists):
|
||||
elif ctype == 'float':
|
||||
x = float(x)
|
||||
elif ctype == 'none':
|
||||
assert x == 'None', 'for none type, the value must be None instead of {:}'.format(x)
|
||||
if x.lower() != 'none':
|
||||
raise ValueError('For the none type, the value must be none instead of {:}'.format(x))
|
||||
x = None
|
||||
else:
|
||||
raise TypeError('Does not know this type : {:}'.format(ctype))
|
||||
|
Reference in New Issue
Block a user