Skip to content
  • This project
    • Loading...
  • Sign in

周伟奇 / test_on_pytorch

Go to a project
Toggle navigation
Toggle navigation pinning
  • Projects
  • Groups
  • Snippets
  • Help
  • Project
  • Activity
  • Repository
  • Pipelines
  • Graphs
  • Issues 0
  • Merge Requests 0
  • Wiki
  • Network
  • Create a new issue
  • Builds
  • Commits
  • Issue Boards
  • Files
  • Commits
  • Network
  • Compare
  • Branches
  • Tags
Switch branch/tag
  • test_on_pytorch
  • solver
  • builder.py
  • 周伟奇's avatar
    first commit · bcb17d0f
    周伟奇 committed 2022-12-01 19:10:09 +0800
    bcb17d0f
builder.py 310 Bytes
Raw Blame History Permalink
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
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)