From f9e977be70ad52a5321e86c21871fa3979160cc6 Mon Sep 17 00:00:00 2001 From: JeongYeon Nam Date: Wed, 4 Jan 2023 23:01:08 +0900 Subject: [PATCH] auxiliary_loss works for Deformable Detr (#20959) fix: auxiliary_loss works Co-authored-by: Jeongyeon Nam --- .../models/deformable_detr/modeling_deformable_detr.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/transformers/models/deformable_detr/modeling_deformable_detr.py b/src/transformers/models/deformable_detr/modeling_deformable_detr.py index 5076fa6417d..e6766782cce 100755 --- a/src/transformers/models/deformable_detr/modeling_deformable_detr.py +++ b/src/transformers/models/deformable_detr/modeling_deformable_detr.py @@ -1965,9 +1965,6 @@ class DeformableDetrForObjectDetection(DeformableDetrPreTrainedModel): outputs_loss["logits"] = logits outputs_loss["pred_boxes"] = pred_boxes if self.config.auxiliary_loss: - intermediate = outputs.intermediate_hidden_states if return_dict else outputs[4] - outputs_class = self.class_embed(intermediate) - outputs_coord = self.bbox_embed(intermediate).sigmoid() auxiliary_outputs = self._set_aux_loss(outputs_class, outputs_coord) outputs_loss["auxiliary_outputs"] = auxiliary_outputs if self.config.two_stage: