From 382f02a4e4c28e7b284ee2f2b7d39f9b271f173a Mon Sep 17 00:00:00 2001 From: eqy Date: Thu, 25 Oct 2018 12:02:19 -0700 Subject: [PATCH 1/2] fix typo in resnet definition --- python/tvm/relay/testing/resnet.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/tvm/relay/testing/resnet.py b/python/tvm/relay/testing/resnet.py index cb489f6e2471..42ad2e7c5a6e 100644 --- a/python/tvm/relay/testing/resnet.py +++ b/python/tvm/relay/testing/resnet.py @@ -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') From 0c91a3e3d15b6184854cfd6daff74c53c443a90f Mon Sep 17 00:00:00 2001 From: eqy Date: Thu, 25 Oct 2018 13:44:53 -0700 Subject: [PATCH 2/2] trigger CI --- python/tvm/relay/testing/resnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/testing/resnet.py b/python/tvm/relay/testing/resnet.py index 42ad2e7c5a6e..9ba57ae09ef5 100644 --- a/python/tvm/relay/testing/resnet.py +++ b/python/tvm/relay/testing/resnet.py @@ -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