import copy from utils.registery import SOLVER_REGISTRY from .mlp_solver import MLPSolver def build_solver(cfg): cfg = copy.deepcopy(cfg) try: solver_cfg = cfg['solver'] except Exception: raise 'should contain {solver}!' return SOLVER_REGISTRY.get(solver_cfg['name'])(cfg)