No public description

PiperOrigin-RevId: 723962946
tensorflow · Feb 6, 2025 · d31fac8 · d31fac8
1 parent 17bf360
commit d31fac8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/official/projects/detr/tasks/detection.py b/official/projects/detr/tasks/detection.py
@@ -149,9 +149,9 @@ def _compute_cost(self, cls_outputs, box_outputs, cls_targets, box_targets):
     total_cost = cls_cost + box_cost + giou_cost
 
     max_cost = (
-        self._task_config.losses.lambda_cls * 0.0
+        self._task_config.losses.lambda_cls * 1.0
         + self._task_config.losses.lambda_box * 4.0
-        + self._task_config.losses.lambda_giou * 0.0
+        + self._task_config.losses.lambda_giou * 1.0
     )
 
     # Set pads to large constant