correct conv2d workload for resnet18 (#750)

29226a5f · Yizhi Liu · Tianqi Chen · a407ec15 · 29226a5f
Commit 29226a5f authored Jan 03, 2018 by Yizhi Liu Committed by Tianqi Chen Jan 03, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

topi/tests/python/test_topi_conv2d_nchw.py
+1 -1

No files found.
--- a/topi/tests/python/test_topi_conv2d_nchw.py
+++ b/topi/tests/python/test_topi_conv2d_nchw.py
@@ -56,7 +56,7 @@ def verify_conv2d_nchw(batch, in_channel, in_size, num_filter, kernel, stride, p

 def test_conv2d_nchw():
    # ResNet18 worklaods
-    verify_conv2d_nchw(1, 3, 224, 64, 7, 3, 2)
+    verify_conv2d_nchw(1, 3, 224, 64, 7, 2, 3)
    verify_conv2d_nchw(1, 64, 56, 64, 3, 1, 1)
    verify_conv2d_nchw(1, 64, 56, 64, 1, 1, 0)
    verify_conv2d_nchw(1, 64, 56, 128, 3, 2, 1)