update

ZhangGongjie · Apr 1, 2022 · aaf1936 · aaf1936
1 parent 5f9f45c
commit aaf1936
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/fast_detr.py b/models/fast_detr.py
@@ -138,7 +138,7 @@ def forward(self, samples: NestedTensor):
 
         out = {'pred_logits': outputs_class[-1], 'pred_boxes': outputs_coords[-1]}
         if self.aux_loss:
-           out['aux_outputs'] = self._set_aux_loss(outputs_class, outputs_coords)
+            out['aux_outputs'] = self._set_aux_loss(outputs_class, outputs_coords)
 
         return out