stable_pretraining.losses

NTXEntLoss([temperature])	Normalized temperature-scaled cross entropy loss.
NegativeCosineSimilarity(*args, **kwargs)	Negative cosine similarity objective.
VICRegLoss([sim_coeff, std_coeff, ...])	SSL objective used in VICReg [Bardes et al., 2021].
BarlowTwinsLoss([lambd])	SSL objective used in Barlow Twins [Zbontar et al., 2021].