unilab.algos.torch.common.device

Functions

get_env_dims(env_name[, sim_backend, ...])

Get (actor_obs_dim, action_dim, critic_obs_dim) from environment.
unilab.algos.torch.common.device.get_env_dims(env_name, sim_backend='mujoco', env_cfg_override=None)[source]

Get (actor_obs_dim, action_dim, critic_obs_dim) from environment.

Parameters:
Return type:

tuple[int, int, int]