def load(opt): model_opt = filter_opt(opt, 'model') model_name = model_opt['model_name'] del model_opt['model_name'] return get_model(model_name, model_opt)
def load(opt): """ Load the model for the options """ model_opt = filter_opt(opt, 'model') model_name = model_opt['model_name'] del model_opt['model_name'] return get_model(model_name, model_opt)
def load(opt, key='model_name'): model_opt = filter_opt(opt, 'model') model_name = model_opt[key] del model_opt[key] return get_model(model_name, model_opt)