megatron.data.autoaugment.SubPolicy

class megatron.data.autoaugment.SubPolicy(operation1, probability1, magnitude_idx1, operation2, probability2, magnitude_idx2, fillcolor)

Bases: object

Definition of a SubPolicy.

A SubPolicy consists of two augmentation operations, each of those parametrized as operation, probability, magnitude. The two operations are applied sequentially on the image upon call.

__call__(img)

Define call method for SubPolicy class.