Skip to content

Commit f5adc1e

Browse files
committed
assert for simplified conv api
1 parent 05944f9 commit f5adc1e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tensorlayer/layers.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1710,7 +1710,7 @@ def Conv2d(net, n_filter=32, filter_size=(3, 3), strides=(1, 1), act = None,
17101710
>>> conv2 = Conv2d(conv2, 128, (3, 3), act=tf.nn.relu, padding='SAME', W_init=w_init, b_init=b_init, name='conv2_2')
17111711
>>> pool2 = MaxPool2d(conv2, (2, 2), padding='SAME', name='pool2')
17121712
"""
1713-
# assert len(strides) == 2, print("len(strides)==2, Conv2d and Conv2dLayer are different.")
1713+
assert len(strides) == 2, "len(strides) should be 2, Conv2d and Conv2dLayer are different."
17141714
if act is None:
17151715
act = tf.identity
17161716
net = Conv2dLayer(net,
@@ -1742,7 +1742,7 @@ def DeConv2d(net, n_out_channel = 32, filter_size=(3, 3),
17421742
act : None or activation function.
17431743
others : see :class:`DeConv2dLayer`.
17441744
"""
1745-
# assert len(strides) == 2, print("len(strides)==2, DeConv2d and DeConv2dLayer are different.")
1745+
assert len(strides) == 2, "len(strides) should be 2, DeConv2d and DeConv2dLayer are different."
17461746
if act is None:
17471747
act = tf.identity
17481748
if batch_size is None:
@@ -1772,7 +1772,7 @@ def MaxPool2d(net, filter_size=(2, 2), strides=None, padding='SAME', name='maxpo
17721772
"""
17731773
if strides is None:
17741774
strides = filter_size
1775-
# assert len(strides) == 2, print("len(strides)==2, MaxPool2d and PoolLayer are different.")
1775+
assert len(strides) == 2, "len(strides) should be 2, MaxPool2d and PoolLayer are different."
17761776
net = PoolLayer(net, ksize=[1, filter_size[0], filter_size[1], 1],
17771777
strides=[1, strides[0], strides[1], 1],
17781778
padding=padding,
@@ -1792,7 +1792,7 @@ def MeanPool2d(net, filter_size=(2, 2), strides=None, padding='SAME', name='mean
17921792
"""
17931793
if strides is None:
17941794
strides = filter_size
1795-
# assert len(strides) == 2, print("len(strides)==2, MeanPool2d and PoolLayer are different.")
1795+
assert len(strides) == 2, "len(strides) should be 2, MeanPool2d and PoolLayer are different."
17961796
net = PoolLayer(net, ksize=[1, filter_size[0], filter_size[1], 1],
17971797
strides=[1, strides[0], strides[1], 1],
17981798
padding=padding,

0 commit comments

Comments
 (0)