set segmentation loss weight to 0.5

official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py

@@ -107,7 +107,7 @@ class Losses(maskrcnn.Losses):
   semantic_segmentation_use_groundtruth_dimension: bool = True
   semantic_segmentation_top_k_percent_pixels: float = 1.0
   instance_segmentation_weight: float = 1.0
-  semantic_segmentation_weight: float = 1.0
+  semantic_segmentation_weight: float = 0.5
 
 
 @dataclasses.dataclass
@@ -170,7 +170,8 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig:
     is_thing.append(True if idx <= num_thing_categories else False)
 
   config = cfg.ExperimentConfig(
-      runtime=cfg.RuntimeConfig(mixed_precision_dtype='bfloat16'),
+    runtime=cfg.RuntimeConfig(
+      mixed_precision_dtype='bfloat16', enable_xla=True),
       task=PanopticMaskRCNNTask(
           init_checkpoint='gs://cloud-tpu-checkpoints/vision-2.0/resnet50_imagenet/ckpt-28080',  # pylint: disable=line-too-long
           init_checkpoint_modules=['backbone'],