Hi @ptrblck, Thanks for the reply. My problem here is SupConLoss, which is defined for Classification problems takes in features of shape 3D, [bsz, n_views, …], and labels of shape 1D, [bsz], but my Segmentation model gives output features of shape 4D, [bsz, feat_dim, h, w] and labels are of shape 3D, [bsz, h, w]. Is there anyway I can convert this loss so that it fits for my output features shape & labels shape.
Thank you