add attention layer and AITM model

alibaba · Jun 13, 2024 · 6683f7d · 6683f7d
1 parent c1da79f
commit 6683f7d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/easy_rec/python/model/multi_task_model.py b/easy_rec/python/model/multi_task_model.py
@@ -301,6 +301,8 @@ def get_outputs(self):
                 suffix='_%s' % tower_name))
       else:
         for loss in task_tower_cfg.losses:
+          if loss.loss_type == LossType.ORDER_CALIBRATE_LOSS:
+            continue
           outputs.extend(
               self._get_outputs_impl(
                   loss.loss_type,