Skip to content

Commit 955eaa2

Browse files
committed
Merge pull request #55 from elPistolero/conv_stride_fix
do stride assertion on the expanded version
2 parents 0327ad5 + 7a52373 commit 955eaa2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

DeepFried2/layers/SpatialConvolution.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ def __init__(self, nchan_in, nchan_out, filter_size, stride=1, border='valid', w
1414
self.stride = expand(stride, len(filter_size), 'stride')
1515
self.border = expand(border, len(filter_size), 'border')
1616

17-
if len(self.filter_size) == 3 and any(s != 1 for s in stride):
17+
if len(self.filter_size) == 3 and any(s != 1 for s in self.stride):
1818
raise NotImplementedError('stride != 1 is not implemented for 3D convolutions')
1919

2020
if len(self.filter_size) == 3 and imshape is not None:

0 commit comments

Comments
 (0)