From 5d755898fdc927bc64228317a6d9c7c382bcdb4e Mon Sep 17 00:00:00 2001 From: P2Oileen Date: Thu, 6 Jan 2022 16:15:08 +0800 Subject: [PATCH] fix(funtional): correct the typo: witdth to width --- imperative/python/megengine/functional/nn.py | 8 ++++---- imperative/python/megengine/module/padding.py | 6 +++--- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/imperative/python/megengine/functional/nn.py b/imperative/python/megengine/functional/nn.py index 9ababceb..c4ff7a31 100644 --- a/imperative/python/megengine/functional/nn.py +++ b/imperative/python/megengine/functional/nn.py @@ -1754,7 +1754,7 @@ def sliding_window_transpose( def pad( src: Tensor, - pad_witdth: Tuple[Tuple[int, int], ...], + pad_width: Tuple[Tuple[int, int], ...], mode: str = "constant", constant_value: float = 0.0, ) -> Tensor: @@ -1769,9 +1769,9 @@ def pad( if mode.lower() == "edge": mode = "replicate" - for i in range(0, len(pad_witdth)): - p_offsets[i * 2] = pad_witdth[i][0] - p_offsets[i * 2 + 1] = pad_witdth[i][1] + for i in range(0, len(pad_width)): + p_offsets[i * 2] = pad_width[i][0] + p_offsets[i * 2 + 1] = pad_width[i][1] op = builtin.Padding( front_offset_dim0=p_offsets[0], diff --git a/imperative/python/megengine/module/padding.py b/imperative/python/megengine/module/padding.py index 433ecd32..078daabc 100644 --- a/imperative/python/megengine/module/padding.py +++ b/imperative/python/megengine/module/padding.py @@ -12,16 +12,16 @@ class Pad(Module): def __init__( self, - pad_witdth: Tuple[Tuple[int, int], ...], + pad_width: Tuple[Tuple[int, int], ...], mode: str = "constant", constant_val: float = 0.0, ): super().__init__() - self.pad_width = pad_witdth + self.pad_width = pad_width self.mode = mode self.pad_val = constant_val def forward(self, src): return nn.pad( - src, pad_witdth=self.pad_width, mode=self.mode, constant_value=self.pad_val + src, pad_width=self.pad_width, mode=self.mode, constant_value=self.pad_val )