d3rlpy.augmentation.image.VerticalFlip¶
-
class
d3rlpy.augmentation.image.
VerticalFlip
(probability=0.1)[source]¶ Vertical flip augmentation.
References
- Parameters
probability (float) – probability to flip vertically.
Methods
-
transform
(x)[source]¶ Returns augmented observation.
- Parameters
x (torch.Tensor) – observation.
- Returns
augmented observation.
- Return type
torch.Tensor
Attributes