osc.viz.loss_global.viz_contrastive_loss_global_probs

viz_contrastive_loss_global_probs(global_feats, temp, loss)[source]

Visualize global contrastive loss between 2B images, probabilities only.

Parameters
  • global_feats (Tensor) – a tensor with shape [2B, C]

  • temp (float) – loss temperature

  • loss (float) – loss value to show on the plot

Return type

Figure

Returns

A figure with a single plot of matching probabilities.