osc.viz.loss_global.viz_contrastive_loss_global

viz_contrastive_loss_global(global_feats, loss)

Visualize global contrastive loss between 2B images.

Parameters

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

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

Return type

Figure

Returns

A figure with two axes.