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

previous

cerebras.modelzoo.losses.clip_contrasive_loss

next

cerebras.modelzoo.losses.dpr_loss