cerebras.modelzoo.losses.clip_contrasive_loss.ClipContrastiveLoss#
- class cerebras.modelzoo.losses.clip_contrasive_loss.ClipContrastiveLoss(*args, **kwargs)[source]#
Bases:
torch.nn.Module
Methods
forward
generate_labels
Bases: torch.nn.Module
Methods
|
|
|