Skip to content

GitLab

"symphony/git@developer.sourcefind.cn:cnjsdfcy/simbricks.git" did not exist on "be8fda17cceee5a95e392e523bd929846a8887a0"
Unverified Commit 170eb70b authored by srihari-humbarwadi's avatar srihari-humbarwadi
Browse files

fix runtime dtype

parent a7b55e81
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py
View file @ 170eb70b
...@@ -181,7 +181,7 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig: ...@@ -181,7 +181,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_resnet50_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'],
......
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