riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

auxiliary_loss works for Deformable Detr (#20959)

Browse Source 
fix: auxiliary_loss works

Co-authored-by: Jeongyeon Nam <jy.nam@navercorp.com>
This commit is contained in:
JeongYeon Nam 2023-01-04 23:01:08 +09:00 committed by GitHub
parent b493fee958
commit f9e977be70
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/deformable_detr/modeling_deformable_detr.py
View File

@ -1965,9 +1965,6 @@ class DeformableDetrForObjectDetection(DeformableDetrPreTrainedModel):
outputs_loss["logits"] = logits outputs_loss["logits"] = logits
outputs_loss["pred_boxes"] = pred_boxes outputs_loss["pred_boxes"] = pred_boxes
if self.config.auxiliary_loss: 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) auxiliary_outputs = self._set_aux_loss(outputs_class, outputs_coord)
outputs_loss["auxiliary_outputs"] = auxiliary_outputs outputs_loss["auxiliary_outputs"] = auxiliary_outputs
if self.config.two_stage: if self.config.two_stage: