unilab.training.rsl_rl.get_policy_obs_dims

unilab.training.rsl_rl.get_policy_obs_dims(obs_groups_spec)[source]

Return (actor_obs_dim, flat_policy_obs_dim) for RSL-RL policies.

Parameters:

obs_groups_spec (dict[str, int])

Return type:

tuple[int, int]