set `pool_kernel_size` and `dropout_rate` for `ASPP`

14f4c9df · srihari-humbarwadi · adf0c8e9 · 14f4c9df
Unverified Commit 14f4c9df authored Mar 30, 2022 by srihari-humbarwadi
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py ...ta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py +3 -1

No files found.
--- a/official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py
+++ b/official/vision/beta/projects/panoptic_maskrcnn/configs/panoptic_deeplab.py
@@ -199,7 +199,9 @@ def panoptic_deeplab_coco() -> cfg.ExperimentConfig:
                  aspp=decoders.ASPP(
                      level=level,
                      num_filters=256,
-                      dilation_rates=aspp_dilation_rates)),
+                      pool_kernel_size=input_size[:2],
+                      dilation_rates=aspp_dilation_rates,
+                      dropout_rate=0.1)),
              semantic_head=SemanticHead(
                  level=level,
                  num_convs=1,