Reformulate via black
This commit is contained in:
		| @@ -5,35 +5,37 @@ | ||||
| ################################################################################################ | ||||
| import sys, argparse | ||||
| from pathlib import Path | ||||
| lib_dir = (Path(__file__).parent / '..' / '..' / 'lib').resolve() | ||||
| if str(lib_dir) not in sys.path: sys.path.insert(0, str(lib_dir)) | ||||
| from nas_201_api  import NASBench201API as API | ||||
|  | ||||
| if __name__ == '__main__': | ||||
|   parser = argparse.ArgumentParser("Analysis of NAS-Bench-201") | ||||
|   parser.add_argument('--api_path',  type=str, default=None,                                         help='The path to the NAS-Bench-201 benchmark file.') | ||||
|   args = parser.parse_args() | ||||
| lib_dir = (Path(__file__).parent / ".." / ".." / "lib").resolve() | ||||
| if str(lib_dir) not in sys.path: | ||||
|     sys.path.insert(0, str(lib_dir)) | ||||
| from nas_201_api import NASBench201API as API | ||||
|  | ||||
|   meta_file = Path(args.api_path) | ||||
|   assert meta_file.exists(), 'invalid path for api : {:}'.format(meta_file) | ||||
| if __name__ == "__main__": | ||||
|     parser = argparse.ArgumentParser("Analysis of NAS-Bench-201") | ||||
|     parser.add_argument("--api_path", type=str, default=None, help="The path to the NAS-Bench-201 benchmark file.") | ||||
|     args = parser.parse_args() | ||||
|  | ||||
|   api = API(str(meta_file)) | ||||
|     meta_file = Path(args.api_path) | ||||
|     assert meta_file.exists(), "invalid path for api : {:}".format(meta_file) | ||||
|  | ||||
|   # This will show the results of the best architecture based on the validation set of each dataset. | ||||
|   arch_index, accuracy = api.find_best('cifar10-valid', 'x-valid', None, None, False) | ||||
|   print('FOR CIFAR-010, using the hyper-parameters with 200 training epochs :::') | ||||
|   print('arch-index={:5d}, arch={:}'.format(arch_index, api.arch(arch_index))) | ||||
|   api.show(arch_index) | ||||
|   print('') | ||||
|     api = API(str(meta_file)) | ||||
|  | ||||
|   arch_index, accuracy = api.find_best('cifar100', 'x-valid', None, None, False) | ||||
|   print('FOR CIFAR-100, using the hyper-parameters with 200 training epochs :::') | ||||
|   print('arch-index={:5d}, arch={:}'.format(arch_index, api.arch(arch_index))) | ||||
|   api.show(arch_index) | ||||
|   print('') | ||||
|     # This will show the results of the best architecture based on the validation set of each dataset. | ||||
|     arch_index, accuracy = api.find_best("cifar10-valid", "x-valid", None, None, False) | ||||
|     print("FOR CIFAR-010, using the hyper-parameters with 200 training epochs :::") | ||||
|     print("arch-index={:5d}, arch={:}".format(arch_index, api.arch(arch_index))) | ||||
|     api.show(arch_index) | ||||
|     print("") | ||||
|  | ||||
|   arch_index, accuracy = api.find_best('ImageNet16-120', 'x-valid', None, None, False) | ||||
|   print('FOR ImageNet16-120, using the hyper-parameters with 200 training epochs :::') | ||||
|   print('arch-index={:5d}, arch={:}'.format(arch_index, api.arch(arch_index))) | ||||
|   api.show(arch_index) | ||||
|   print('') | ||||
|     arch_index, accuracy = api.find_best("cifar100", "x-valid", None, None, False) | ||||
|     print("FOR CIFAR-100, using the hyper-parameters with 200 training epochs :::") | ||||
|     print("arch-index={:5d}, arch={:}".format(arch_index, api.arch(arch_index))) | ||||
|     api.show(arch_index) | ||||
|     print("") | ||||
|  | ||||
|     arch_index, accuracy = api.find_best("ImageNet16-120", "x-valid", None, None, False) | ||||
|     print("FOR ImageNet16-120, using the hyper-parameters with 200 training epochs :::") | ||||
|     print("arch-index={:5d}, arch={:}".format(arch_index, api.arch(arch_index))) | ||||
|     api.show(arch_index) | ||||
|     print("") | ||||
|   | ||||
		Reference in New Issue
	
	Block a user