from .resnet20 import rf_resnet20
from ..genotypes import * 
from ..spaces import * 
from ..operations import *
from lib.models.autorf import genotypes 


MODEL_LIST = {
    'resnet20': rf_resnet20,
}



def build_auto_network(model_name='resnet20',num_classes=10, genotype=None):
    return MODEL_LIST[model_name](num_classes=num_classes, genotype=genotype)
from .resnet20 import rf_resnet20
from ..genotypes import * 
from ..spaces import * 
from ..operations import *
from lib.models.autorf import genotypes 


MODEL_LIST = {
    'resnet20': rf_resnet20,
}



def build_auto_network(model_name='resnet20',num_classes=10, genotype=None):
    return MODEL_LIST[model_name](num_classes=num_classes, genotype=genotype)