コード例 #1
0
ファイル: registry.py プロジェクト: Simplify1/rl_algorithms
def build_agent(cfg: ConfigDict, build_args: dict = None):
    """Build agent using config and additional arguments."""
    return build_from_cfg(cfg, AGENTS, build_args)
コード例 #2
0
ファイル: registry.py プロジェクト: Simplify1/rl_algorithms
def build_her(cfg: ConfigDict, build_args: dict = None):
    """Build her using config and additional arguments."""
    return build_from_cfg(cfg, HERS, build_args)
コード例 #3
0
def build_loss(cfg: ConfigDict, build_args: dict = None):
    """Build loss using config and additional arguments."""
    return build_from_cfg(cfg, LOSSES, build_args)
コード例 #4
0
ファイル: registry.py プロジェクト: Simplify1/rl_algorithms
def build_backbone(cfg: ConfigDict, build_args: dict = None):
    """Build agent using config and additional arguments."""
    return build_from_cfg(cfg, BACKBONES, build_args)
コード例 #5
0
def build_worker(cfg: ConfigDict, build_args: dict = None):
    """Build ray worker using config and additional arguments."""
    # return build_ray_obj_from_cfg(cfg, WORKERS, build_args)
    return build_from_cfg(cfg, WORKERS, build_args)