iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update CVPR-2019-GDAS re-train NASNet-search-space searched models

Browse Source
This commit is contained in:
D-X-Y
2020-03-06 19:29:07 +11:00
parent 8b6df42f1f
commit 9a83814a46
17 changed files with 278 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
exps/basic-main.py
View File

@@ -39,7 +39,9 @@ def main(args):
if args.model_source == 'normal':
base_model = obtain_model(model_config)
elif args.model_source == 'nas':
base_model = obtain_nas_infer_model(model_config)
base_model = obtain_nas_infer_model(model_config, args.extra_model_path)
elif args.model_source == 'autodl-searched':
base_model = obtain_model(model_config, args.extra_model_path)
else:
raise ValueError('invalid model-source : {:}'.format(args.model_source))
flop, param = get_model_infos(base_model, xshape)