Contents:
Visualize global contrastive loss between 2B images, probabilities only.
global_feats (Tensor) – a tensor with shape [2B, C]
Tensor
temp (float) – loss temperature
float
loss (float) – loss value to show on the plot
Figure
A figure with a single plot of matching probabilities.