Skip to content

Commit cd7b408

Browse files
committed
fix PoolRule
1 parent 44e074c commit cd7b408

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tester/paddle_to_torch/rules.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4469,7 +4469,7 @@ def _get_same_padding_1d(input_size, kernel_size, stride):
44694469
pad_left, pad_right = _get_same_padding_1d(input_size, kernel_size, stride)
44704470
padding = pad_left # 对称填充
44714471
if pad_left != pad_right: # 非对称填充
4472-
x = torch.nn.functional.pad(x, (pad_left, pad_right))
4472+
x = torch.nn.functional.pad(x, (pad_left, pad_right), mode="replicate")
44734473
padding = 0
44744474
elif isinstance(padding, (list, tuple)):
44754475
if len(padding) == 1: # [pad]
@@ -4508,7 +4508,7 @@ def _get_same_padding_2d(input_size, kernel_size, stride):
45084508
pad_h, pad_w = _get_same_padding_2d(input_size, kernel_size, stride)
45094509
padding = (pad_h[0], pad_w[0]) # 对称填充
45104510
if pad_h[0] != pad_h[1] or pad_w[0] != pad_w[1]: # 非对称填充
4511-
x = torch.nn.functional.pad(x, (pad_w[0], pad_w[1], pad_h[0], pad_h[1]))
4511+
x = torch.nn.functional.pad(x, (pad_w[0], pad_w[1], pad_h[0], pad_h[1]), mode="replicate")
45124512
padding = 0
45134513
elif isinstance(padding, (list, tuple)):
45144514
if len(padding) == 2: # [pad_height, pad_width]

0 commit comments

Comments
 (0)