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