Skip to content

GitLab

Unverified Commit 2e908198 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

set segmentation loss weight to 0.5

parent d1f01835
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_maskrcnn.py
View file @ 2e908198
...@@ -107,7 +107,7 @@ class Losses(maskrcnn.Losses): ...@@ -107,7 +107,7 @@ class Losses(maskrcnn.Losses):
semantic_segmentation_use_groundtruth_dimension: bool = True semantic_segmentation_use_groundtruth_dimension: bool = True
semantic_segmentation_top_k_percent_pixels: float = 1.0 semantic_segmentation_top_k_percent_pixels: float = 1.0
instance_segmentation_weight: float = 1.0 instance_segmentation_weight: float = 1.0
semantic_segmentation_weight: float = 1.0 semantic_segmentation_weight: float = 0.5
@dataclasses.dataclass @dataclasses.dataclass
...@@ -170,7 +170,8 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig: ...@@ -170,7 +170,8 @@ def panoptic_fpn_coco() -> cfg.ExperimentConfig:
is_thing.append(True if idx <= num_thing_categories else False) is_thing.append(True if idx <= num_thing_categories else False)
config = cfg.ExperimentConfig( config = cfg.ExperimentConfig(
runtime=cfg.RuntimeConfig(mixed_precision_dtype='bfloat16'), runtime=cfg.RuntimeConfig(
mixed_precision_dtype='bfloat16', enable_xla=True),
task=PanopticMaskRCNNTask( task=PanopticMaskRCNNTask(
init_checkpoint='gs://cloud-tpu-checkpoints/vision-2.0/resnet50_imagenet/ckpt-28080', # pylint: disable=line-too-long init_checkpoint='gs://cloud-tpu-checkpoints/vision-2.0/resnet50_imagenet/ckpt-28080', # pylint: disable=line-too-long
init_checkpoint_modules=['backbone'], init_checkpoint_modules=['backbone'],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment