Skip to content

GitLab

Unverified Commit 95f061ba authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

use `stem_type=v0` for `dilated_resnet`

parent 2597ef4b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py
View file @ 95f061ba
...@@ -173,14 +173,14 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig: ...@@ -173,14 +173,14 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig:
output_stride = 16 output_stride = 16
aspp_dilation_rates = [6, 12, 18] aspp_dilation_rates = [6, 12, 18]
multigrid = [1, 2, 4] multigrid = [1, 2, 4]
stem_type = 'v1' stem_type = 'v0'
level = int(np.math.log2(output_stride)) level = int(np.math.log2(output_stride))
config = cfg.ExperimentConfig( config = cfg.ExperimentConfig(
runtime=cfg.RuntimeConfig( runtime=cfg.RuntimeConfig(
mixed_precision_dtype='bfloat16', enable_xla=True), mixed_precision_dtype='bfloat16', enable_xla=True),
task=PanopticDeeplabTask( task=PanopticDeeplabTask(
init_checkpoint='gs://cloud-tpu-checkpoints/vision-2.0/deeplab/deeplab_resnet101_imagenet/ckpt-62400', # pylint: disable=line-too-long init_checkpoint='gs://cloud-tpu-checkpoints/vision-2.0/deeplab/deeplab_resnet50_imagenet/ckpt-62400', # pylint: disable=line-too-long
init_checkpoint_modules=['backbone'], init_checkpoint_modules=['backbone'],
model=PanopticDeeplab( model=PanopticDeeplab(
num_classes=num_panoptic_categories, num_classes=num_panoptic_categories,
......
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