cerebras.modelzoo.data.vision.diffusion.dit_transforms.LabelDropout#
- class cerebras.modelzoo.data.vision.diffusion.dit_transforms.LabelDropout(*args, **kwargs)[source]#
Bases:
torch.nn.Module
Methods
forward
token_drop
Bases: torch.nn.Module
Methods
|
|
|