iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Create NATS

Browse Source
This commit is contained in:
D-X-Y
2020-07-30 13:07:11 +00:00
parent df45e68366
commit 6061d74631
21 changed files with 1336 additions and 126 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
exps/experimental/visualize-nas-bench-x.py
View File

@@ -21,9 +21,9 @@ import matplotlib.ticker as ticker
lib_dir = (Path(__file__).parent / '..' / '..' / 'lib').resolve()
if str(lib_dir) not in sys.path: sys.path.insert(0, str(lib_dir))
from config_utils import dict2config, load_config
from nas_201_api import NASBench201API, NASBench301API
from log_utils import time_string
from models import get_cell_based_tiny_net
from nats_bench import create
def visualize_info(api, vis_save_dir, indicator):
@@ -391,11 +391,11 @@ if __name__ == '__main__':
to_save_dir = Path(args.save_dir)
datasets = ['cifar10', 'cifar100', 'ImageNet16-120']
api201 = NASBench201API(None, verbose=True)
api201 = create(None, 'tss', verbose=True)
for xdata in datasets:
visualize_tss_info(api201, xdata, to_save_dir)
api301 = NASBench301API(None, verbose=True)
api301 = create(None, 'size', verbose=True)
for xdata in datasets:
visualize_sss_info(api301, xdata, to_save_dir)