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