builder.py
310 Bytes
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)