Modify default value of pad_to_patch_size arg.

open-mmlab · Jul 19, 2021 · e7039ff · e7039ff
1 parent 38b2db8
commit e7039ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmseg/models/utils/embed.py b/mmseg/models/utils/embed.py
@@ -20,7 +20,7 @@ class PatchEmbed(BaseModule):
         padding (int): The padding length of embedding conv. Default: 0.
         dilation (int): The dilation rate of embedding conv. Default: 1.
         pad_to_patch_size (bool, optional): Whether to pad feature map shape
-            to multiple patch size. Default: False.
+            to multiple patch size. Default: True.
         norm_cfg (dict, optional): Config dict for normalization layer.
         init_cfg (`mmcv.ConfigDict`, optional): The Config for initialization.
             Default: None.
@@ -34,7 +34,7 @@ def __init__(self,
                  stride=16,
                  padding=0,
                  dilation=1,
-                 pad_to_patch_size=False,
+                 pad_to_patch_size=True,
                  norm_cfg=None,
                  init_cfg=None):
         super(PatchEmbed, self).__init__()