Commit 449b277

authored

Merge pull request #223 from matkalinowski/static_padding

static padding fixed

2 parents e055a74 + d8c4f73 commit 449b277Copy full SHA for 449b277

File tree

-1

lines changed

-1

lines changed

Lines changed: 2 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -261,7 +261,8 @@ def __init__(self, in_channels, out_channels, kernel_size, stride=1, image_size=`
`261`	`261`	`pad_h = max((oh - 1) * self.stride[0] + (kh - 1) * self.dilation[0] + 1 - ih, 0)`
`262`	`262`	`pad_w = max((ow - 1) * self.stride[1] + (kw - 1) * self.dilation[1] + 1 - iw, 0)`
`263`	`263`	`if pad_h > 0 or pad_w > 0:`
`264`		`- self.static_padding = nn.ZeroPad2d((pad_w // 2, pad_w - pad_w // 2, pad_h // 2, pad_h - pad_h // 2))`
	`264`	`+ self.static_padding = nn.ZeroPad2d((pad_w - pad_w // 2, pad_w - pad_w // 2,`
	`265`	`+ pad_h - pad_h // 2, pad_h - pad_h // 2))`
`265`	`266`	`else:`
`266`	`267`	`self.static_padding = nn.Identity()`
`267`	`268`

Comments

(0)