Skip to content

GitLab

Unverified Commit 177e2d71 authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fix bug in `learning_rate` config

parent f024e381
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py
View file @ 177e2d71
...@@ -178,7 +178,7 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig: ...@@ -178,7 +178,7 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig:
config = cfg.ExperimentConfig( config = cfg.ExperimentConfig(
runtime=cfg.RuntimeConfig( runtime=cfg.RuntimeConfig(
mixed_precision_dtype='float32', 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_resnet101_imagenet/ckpt-62400', # pylint: disable=line-too-long
init_checkpoint_modules=['backbone'], init_checkpoint_modules=['backbone'],
...@@ -282,6 +282,7 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig: ...@@ -282,6 +282,7 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig:
'type': 'adam', 'type': 'adam',
}, },
'learning_rate': { 'learning_rate': {
'type': 'polynomial',
'polynomial': { 'polynomial': {
'initial_learning_rate': 0.0005, 'initial_learning_rate': 0.0005,
'decay_steps': train_steps, 'decay_steps': train_steps,
......
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