Skip to content

Commit 4dacaa0

Browse files
authored
Merge pull request hassony2#5 from rimchang/patch-1
I think it should be ceil_mode=True
2 parents c2b54db + bdbdc77 commit 4dacaa0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/i3dpt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ def __init__(self, kernel_size, stride=None, padding='SAME'):
113113
padding_shape = get_padding_shape(kernel_size, stride)
114114
self.padding_shape = padding_shape
115115
self.pad = torch.nn.ConstantPad3d(padding_shape, 0)
116-
self.pool = torch.nn.MaxPool3d(kernel_size, stride)
116+
self.pool = torch.nn.MaxPool3d(kernel_size, stride, ceil_mode=True)
117117

118118
def forward(self, inp):
119119
inp = self.pad(inp)

0 commit comments

Comments
 (0)