Unverified Commit f9e977be authored by JeongYeon Nam's avatar JeongYeon Nam Committed by GitHub
Browse files

auxiliary_loss works for Deformable Detr (#20959)



fix: auxiliary_loss works
Co-authored-by: default avatarJeongyeon Nam <jy.nam@navercorp.com>
parent b493fee9
...@@ -1965,9 +1965,6 @@ class DeformableDetrForObjectDetection(DeformableDetrPreTrainedModel): ...@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment