From 6683f7d5f3f4c097a15d2fc52a9d15fed78e98a0 Mon Sep 17 00:00:00 2001 From: "weisu.yxd" Date: Thu, 13 Jun 2024 21:01:35 +0800 Subject: [PATCH] add attention layer and AITM model --- easy_rec/python/model/multi_task_model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/easy_rec/python/model/multi_task_model.py b/easy_rec/python/model/multi_task_model.py index 0c5df7ae4..fa6ce8948 100644 --- 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,