Skip to content

Commit c7e285a

Browse files
committed
Update SpatialMaxPooling to new Theano parameter names.
1 parent 8b8a864 commit c7e285a

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

DeepFried2/layers/SpatialMaxPooling.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def symb_forward(self, symb_input):
3636

3737
# downsample height and width first
3838
# other dimensions contribute to batch_size
39-
op = Pool(self.window_size[1:], self.ignore_border, st=self.stride[1:], padding=self.padding[1:])
39+
op = Pool(self.window_size[1:], self.ignore_border, stride=self.stride[1:], pad=self.padding[1:])
4040
output = op(input_4d)
4141

4242
outshape = df.T.join(0, symb_input.shape[:-2], output.shape[-2:])
@@ -56,7 +56,7 @@ def symb_forward(self, symb_input):
5656

5757
# downsample depth
5858
# other dimensions contribute to batch_size
59-
op = Pool((1,self.window_size[0]), self.ignore_border, st=(1,self.stride[0]), padding=(0, self.padding[0]))
59+
op = Pool((1,self.window_size[0]), self.ignore_border, stride=(1,self.stride[0]), pad=(0, self.padding[0]))
6060
outdepth = op(input_4D_depth)
6161

6262
outshape = df.T.join(0, input_depth.shape[:-2], outdepth.shape[-2:])
@@ -66,8 +66,8 @@ def symb_forward(self, symb_input):
6666
else:
6767
return pool_2d(
6868
symb_input,
69-
ds=self.window_size,
69+
ws=self.window_size,
7070
ignore_border=self.ignore_border,
71-
st=self.stride,
72-
padding=self.padding
71+
stride=self.stride,
72+
pad=self.padding
7373
)

0 commit comments

Comments
 (0)