cerebras.modelzoo.data.vision.diffusion.dit_transforms.LabelDropout

class cerebras.modelzoo.data.vision.diffusion.dit_transforms.LabelDropout

Bases: torch.nn.Module

Methods

forward
token_drop

__init__(dropout_prob, num_classes)

__call__(*args: Any, **kwargs: Any) → Any

Call self as a function.

static __new__(cls, *args: Any, **kwargs: Any) → Any

previous

cerebras.modelzoo.data.vision.diffusion.dit_transforms

next

cerebras.modelzoo.data.vision.diffusion.dit_transforms.NoiseGenerator