@@ -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
44744474elif 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
45134513elif isinstance(padding, (list, tuple)):
45144514 if len(padding) == 2: # [pad_height, pad_width]
0 commit comments