chore: import upstream snapshot with attribution

2026-07-13 13:34:58 +08:00
commit a203934033
1368 changed files with 175001 additions and 0 deletions
@@ -0,0 +1,16 @@
+# Copyright (c) ModelScope Contributors. All rights reserved.
+from .causal_lm import CustomCrossEntropyLoss
+from .embedding import ContrastiveLoss, CosineSimilarityLoss, InfonceLoss, OnlineContrastiveLoss
+from .reranker import ListwiseRerankerLoss, PointwiseRerankerLoss
+
+loss_map = {
+    'cross_entropy': CustomCrossEntropyLoss,  # examples
+    # embedding
+    'cosine_similarity': CosineSimilarityLoss,
+    'contrastive': ContrastiveLoss,
+    'online_contrastive': OnlineContrastiveLoss,
+    'infonce': InfonceLoss,
+    # # reranker
+    'pointwise_reranker': PointwiseRerankerLoss,
+    'listwise_reranker': ListwiseRerankerLoss,
+}