unilab.base.augmentation

Classes

SymmetryAugmentation

Runtime symmetry augmentation contract owned by env/backend adapters.
class unilab.base.augmentation.SymmetryAugmentation[source]

Bases: Protocol

Runtime symmetry augmentation contract owned by env/backend adapters.

batch_multiplier: int
augment_obs_and_actions(obs, actions, *, obs_group='obs')[source]
Parameters:
Return type:

tuple[Tensor, Tensor]

mirror_obs(obs, *, obs_group='obs')[source]
Parameters:
Return type:

Tensor

__init__(*args, **kwargs)