Merge pull request #533 from peterzhang2029/resnet_fix

Fix resnet_cifar10 in image_classification.
PaddlePaddle · Dec 11, 2017 · 08d1882 · 08d1882
2 parents ca01557 + ea1ded3
commit 08d1882
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/image_classification/resnet.py b/image_classification/resnet.py
@@ -85,9 +85,9 @@ def resnet_cifar10(input, class_dim, depth=32):
     nStages = {16, 64, 128}
     conv1 = conv_bn_layer(
         input, ch_in=3, ch_out=16, filter_size=3, stride=1, padding=1)
-    res1 = layer_warp(basicblock, conv1, 16, 16, n, 1)
-    res2 = layer_warp(basicblock, res1, 16, 32, n, 2)
-    res3 = layer_warp(basicblock, res2, 32, 64, n, 2)
+    res1 = layer_warp(basicblock, conv1, 16, n, 1)
+    res2 = layer_warp(basicblock, res1, 32, n, 2)
+    res3 = layer_warp(basicblock, res2, 64, n, 2)
     pool = paddle.layer.img_pool(
         input=res3, pool_size=8, stride=1, pool_type=paddle.pooling.Avg())
     out = paddle.layer.fc(