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