Help Needed

[user432]

Hey, Can Supervised Contrastive Loss be implemented for Semantic Segmentation where the output of the model is of the shape;
features = [bsz, feat_dim, h, w]
target = [bsz, h, w]
target_embedding = [bsz, embed_dim, h, w]
It would be very helpful if you can help me. Thanks:)

[HobbitLong]

At least you need to modify the input-output format