from models.search_cnn import SearchCNNController
model = SearchCNNController(3, 16, 10, 20, None, n_nodes=4)
genotype = model.genotype()

python search.py 
--name test 
--data_path /data # path to data 
--dataset CIFAR10 # choose between CIFAR10, CIFAR100, Sport8, MIT67 and flowers102

python augment.py 
--name test 
--layers 20 # 20 for CIFAR10 and CIFAR100, 8 for Sport8, MIT67 and flowers102
--dataset CIFAR10 # choose between CIFAR10, CIFAR100, Sport8, MIT67 and flowers102 
--datapath /data # path to data 
--genotype genotype