提交 79e95c1f 编写于 作者: G gangliao 提交者: GitHub

Merge pull request #1568 from PaddlePaddle/gangliao-patch-1

Set VGG in image classification demo
......@@ -16,7 +16,7 @@ import sys
import paddle.v2 as paddle
from api_v2_resnet import resnet_cifar10
from api_v2_vgg import vgg_bn_drop
def main():
......@@ -31,9 +31,9 @@ def main():
# Add neural network config
# option 1. resnet
net = resnet_cifar10(image, depth=32)
# net = resnet_cifar10(image, depth=32)
# option 2. vgg
# net = vgg_bn_drop(image)
net = vgg_bn_drop(image)
out = paddle.layer.fc(input=net,
size=classdim,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册