Beispiel #1
0
def get_agent_from_checkpoint(checkpoint,
                              agent_name,
                              env=None,
                              **config_kwargs):
    """Instatiate and restore agent class from checkpoint."""
    config = get_config_from_checkpoint(checkpoint, **config_kwargs)
    agent_cls = get_agent_cls(agent_name)

    agent = agent_cls(env=env, config=config)
    agent.restore(checkpoint)
    return agent
Beispiel #2
0
def trainer_cls():
    return get_agent_cls("MBPO")
Beispiel #3
0
def trainer_cls():
    from raylab.agents.registry import get_agent_cls

    return get_agent_cls("SoftAC")
Beispiel #4
0
def svg_trainer(request):
    return get_agent_cls(request.param)
Beispiel #5
0
def svg_inf_trainer():
    return get_agent_cls("SVG(inf)")
Beispiel #6
0
def svg_one_trainer():
    return get_agent_cls("SVG(1)")