Internal change

PiperOrigin-RevId: 440164472

Internal change
PiperOrigin-RevId: 440164472
eb956456 · A. Unique TensorFlower · 510e763c · eb956456
Commit eb956456 authored Apr 07, 2022 by A. Unique TensorFlower
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/vision/tasks/semantic_segmentation.py official/vision/tasks/semantic_segmentation.py +1 -1

No files found.
--- a/official/vision/tasks/semantic_segmentation.py
+++ b/official/vision/tasks/semantic_segmentation.py
@@ -166,7 +166,7 @@ class SemanticSegmentationTask(base_task.Task):
      **kwargs: other args.
    """
    for metric in metrics:
-      if 'mask_scores_mse' is metric.name:
+      if 'mask_scores_mse' == metric.name:
        actual_mask_scores = segmentation_losses.get_actual_mask_scores(
            model_outputs['logits'], labels['masks'],
            self.task_config.losses.ignore_label)