Language Language English 简体中文

unilab.envs.locomotion.g1.symmetry.G1SymmetryAugmentation

class unilab.envs.locomotion.g1.symmetry.G1SymmetryAugmentation

Bases: SymmetryAugmentation

Runtime symmetry adapter derived from the MuJoCo actuator ordering.

Parameters:

• obs_layouts (dict[str, tuple[tuple[str, int], ...]])

• device (str)

Methods

__init__(model, obs_layouts, *[, device])
augment_obs_and_actions(obs, actions, *[, ...])
mirror_action(action)
mirror_obs(obs, *[, obs_group])

Attributes

batch_multiplier

batch_multiplier: int = 2

__init__(model, obs_layouts, *, device='cuda')

Parameters:

• obs_layouts (dict[str, tuple[tuple[str, int], ...]])

• device (str)

mirror_action(action)

Parameters:

action (Tensor)

Return type:

Tensor

mirror_obs(obs, *, obs_group='obs')

Parameters:

• obs (Tensor)

• obs_group (str)

Return type:

Tensor

augment_obs_and_actions(obs, actions, *, obs_group='obs')

Parameters:

• obs (Tensor)

• actions (Tensor)

• obs_group (str)

Return type:

tuple[Tensor, Tensor]