opt_list = [DARTSOptimizer, GDASOptimizer, PCDARTSOptimizer]
parser = argparse.ArgumentParser('nasbench201')
parser.add_argument('--optimizer', type=str, default='DARTSOptimizer')
parser.add_argument('--seed', type=int, default=1)
parser.add_argument('--dataset', type=str, default='cifar10')
parser.add_argument('--epochs',
type=int,
default=50,
help='num of training epochs')
parser.add_argument('--config', type=str, default='final_eval')
args = parser.parse_args()
if __name__ == '__main__':
config = config_parser('../../configs/default.yaml')
parser = Parser('../../configs/default.yaml')
config.seed = parser.config.seed = args.seed
config.dataset = parser.config.dataset = args.dataset
parser.config.save += '/{}/{}'.format(args.optimizer, args.dataset)
create_exp_dir(parser.config.save)
search_space = MacroGraph.from_config(config=config,
filename=os.path.join(
parser.config.save,
'graph.yaml'),
ops_dict=OPS)
# discretize
config = config_parser('../../configs/{}.yaml'.format(args.config))
parser = Parser('../../configs/{}.yaml'.format(args.config))
parser.add_argument('--dataset', type=str, default='cifar10')
parser.add_argument('--space', type=str, default='s1')
parser.add_argument('--epochs',
type=int,
default=50,
help='num of training epochs')
args = parser.parse_args()
robust_darts_primitives = {
's2': ['skip_connect', 'sep_conv_3x3'],
's3': ['none', 'skip_connect', 'sep_conv_3x3'],
's4': ['noise', 'sep_conv_3x3']
}
if __name__ == '__main__':
config = config_parser('../../configs/default.yaml')
parser = Parser('../../configs/default.yaml')
config.seed = parser.config.seed = args.seed
config.dataset = parser.config.dataset = args.dataset
parser.config.save += '/{}/{}/{}'.format(args.optimizer, args.dataset,
args.space)
create_exp_dir(parser.config.save)
fh = logging.FileHandler(
os.path.join(parser.config.save, 'log_{}.txt'.format(config.seed)))
fh.setFormatter(logging.Formatter(log_format))
logging.getLogger().addHandler(fh)
one_shot_optimizer = eval(args.optimizer).from_config(**config)
if args.space == 's1':
search_space = MacroGraph.from_config(
opt_list = [DARTSOptimizer, GDASOptimizer, PCDARTSOptimizer]
log_format = '%(asctime)s %(message)s'
logging.basicConfig(stream=sys.stdout, level=logging.INFO,
format=log_format, datefmt='%m/%d %I:%M:%S %p')
parser = argparse.ArgumentParser('nasbench201')
parser.add_argument('--optimizer', type=str, default='DARTSOptimizer')
parser.add_argument('--seed', type=int, default=1)
parser.add_argument('--dataset', type=str, default='cifar10')
parser.add_argument('--epochs', type=int, default=50, help='num of training epochs')
args = parser.parse_args()
if __name__ == '__main__':
config = config_parser('../../configs/nasbench_201.yaml')
parser = Parser('../../configs/nasbench_201.yaml')
config.seed = parser.config.seed = args.seed
config.epochs = parser.config.epochs = args.epochs
parser.config.save += '/{}'.format('SDARTS' + args.optimizer)
create_exp_dir(parser.config.save)
fh = logging.FileHandler(os.path.join(parser.config.save,
'log_{}.txt'.format(config.seed)))
fh.setFormatter(logging.Formatter(log_format))
logging.getLogger().addHandler(fh)
one_shot_optimizer = eval(args.optimizer).from_config(**config)
search_space = MacroGraph.from_optimizer_op(
one_shot_optimizer,
config=config,
