Skip to content

Commit d31fac8

Browse files
No public description
PiperOrigin-RevId: 723962946
1 parent 17bf360 commit d31fac8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: official/projects/detr/tasks/detection.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -149,9 +149,9 @@ def _compute_cost(self, cls_outputs, box_outputs, cls_targets, box_targets):
149149
total_cost = cls_cost + box_cost + giou_cost
150150

151151
max_cost = (
152-
self._task_config.losses.lambda_cls * 0.0
152+
self._task_config.losses.lambda_cls * 1.0
153153
+ self._task_config.losses.lambda_box * 4.0
154-
+ self._task_config.losses.lambda_giou * 0.0
154+
+ self._task_config.losses.lambda_giou * 1.0
155155
)
156156

157157
# Set pads to large constant

0 commit comments

Comments
 (0)