Commit 44df04a1 by eqy Committed by Tianqi Chen

fix typo in resnet definition (#1995)

parent 0f7aa30b
......@@ -60,7 +60,7 @@ def residual_unit(data,
bn1 = layers.batch_norm_infer(data=data,
epsilon=2e-5,
name=name + '_bn1')
act1 = relay.relu(data=bn1)
act1 = relay.nn.relu(data=bn1)
conv1 = layers.conv2d(
data=act1,
channels=int(num_filter*0.25),
......@@ -69,12 +69,12 @@ def residual_unit(data,
padding=(0, 0),
name=name + '_conv1')
bn2 = layers.batch_norm_infer(data=conv1, epsilon=2e-5, name=name + '_bn2')
act2 = relay.relu(data=bn2)
act2 = relay.nn.relu(data=bn2)
conv2 = layers.conv2d(
data=act2, channels=int(num_filter*0.25), kernel_size=(3, 3),
strides=(1, 1), padding=(1, 1), name=name + '_conv2')
bn3 = layers.batch_norm_infer(data=conv2, epsilon=2e-5, name=name + '_bn3')
act3 = relay.relu(data=bn3)
act3 = relay.nn.relu(data=bn3)
conv3 = layers.conv2d(
data=act3, channels=num_filter, kernel_size=(1, 1),
strides=(1, 1), padding=(0, 0), name=name + '_conv3')
......@@ -120,7 +120,7 @@ def resnet(units,
Number of units in each stage
num_stages : int
Number of stage
Number of stages
filter_list : list
Channel size of each stage
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment