Skip to content
This repository was archived by the owner on Jan 2, 2021. It is now read-only.

Commit c8e8056

Browse files
committed
Fix for padding in convolution.
1 parent 2a29fc9 commit c8e8056

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

enhance.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -268,7 +268,7 @@ def last_layer(self):
268268
return list(self.network.values())[-1]
269269

270270
def make_layer(self, name, input, units, filter_size=(3,3), stride=(1,1), pad=(1,1), alpha=0.25):
271-
conv = ConvLayer(input, units, filter_size, stride=stride, pad=(0,0), nonlinearity=None)
271+
conv = ConvLayer(input, units, filter_size, stride=stride, pad=pad, nonlinearity=None)
272272
prelu = lasagne.layers.ParametricRectifierLayer(conv, alpha=lasagne.init.Constant(alpha))
273273
self.network[name+'x'] = conv
274274
self.network[name+'>'] = prelu

0 commit comments

Comments
 (0)