nnrl/nn/model/stochastic/builders.py
def build_ensemble(obs_space: Box, action_space: Box, spec: EnsembleSpec) -> SME:
    """Construct stochastic dynamics model ensemble.

    Args:
        obs_space: Observation space
        action_space: Action space
        spec: Specifications for stochastic dynamics model ensemble

    Returns:
        A stochastic dynamics model ensemble
    """
    models = [build(obs_space, action_space, spec) for _ in range(spec.ensemble_size)]
    cls = ForkedSME if spec.parallelize else SME
    ensemble = cls(models)
    return ensemble
Construct stochastic dynamics model ensemble.

    Args:
        obs_space: Observation space
        action_space: Action space
        spec: Specifications for stochastic dynamics model ensemble

    Returns:
        A stochastic dynamics model ensemble
    
Construct stochastic dynamics model ensemble.
Construct stochastic dynamics model ensemble .
def build_ensemble(obs_space: Box, action_space: Box, spec: EnsembleSpec) -> SME:
    models = [build(obs_space, action_space, spec) for _ in range(spec.ensemble_size)]
    cls = ForkedSME if spec.parallelize else SME
    ensemble = cls(models)
    return ensemble
def build_ensemble ( obs_space : Box , action_space : Box , spec : EnsembleSpec ) -> SME : models = [ build ( obs_space , action_space , spec ) for _ in range ( spec . ensemble_size ) ] cls = ForkedSME if spec . parallelize else SME ensemble = cls ( models ) return ensemble
Construct stochastic dynamics model ensemble.
Construct stochastic dynamics model ensemble .
"""Construct stochastic dynamics model ensemble.

    Args:
        obs_space: Observation space
        action_space: Action space
        spec: Specifications for stochastic dynamics model ensemble

    Returns:
        A stochastic dynamics model ensemble
    """
A stochastic dynamics model ensemble
A stochastic dynamics model ensemble
obs_space
Box
None
action_space
Box
None
spec
EnsembleSpec
Specifications for stochastic dynamics model ensemble
Specifications for stochastic dynamics model ensemble