def obtain_model(config, extra_path=None):
    if (config.dataset == 'cifar'):
        return get_cifar_models(config, extra_path)
    elif (config.dataset == 'imagenet'):
        return get_imagenet_models(config)
    else:
        raise ValueError('invalid dataset in the model config : {:}'.format(config))
def obtain_model(config, extra_path=None):
    if (config.dataset == 'cifar'):
        return get_cifar_models(config, extra_path)
    elif (config.dataset == 'imagenet'):
        return get_imagenet_models(config)
    else:
        raise ValueError('invalid dataset in the model config : {:}'.format(config))