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,