Browse Source

fix(funtional): correct the typo: witdth to width

tags/v1.9.0
P2Oileen 3 years ago
parent
commit
5d755898fd
2 changed files with 7 additions and 7 deletions
  1. +4
    -4
      imperative/python/megengine/functional/nn.py
  2. +3
    -3
      imperative/python/megengine/module/padding.py

+ 4
- 4
imperative/python/megengine/functional/nn.py View File

@@ -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],


+ 3
- 3
imperative/python/megengine/module/padding.py View File

@@ -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
)

Loading…
Cancel
Save