From 929a390f6c14821be9b4dafd5d150225a23b8cf6 Mon Sep 17 00:00:00 2001 From: gangliao Date: Tue, 7 Mar 2017 11:24:32 +0800 Subject: [PATCH] Set VGG in image classification demo --- demo/image_classification/api_v2_train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/demo/image_classification/api_v2_train.py b/demo/image_classification/api_v2_train.py index 7134fa61e86..53cffa6fb4e 100644 --- a/demo/image_classification/api_v2_train.py +++ b/demo/image_classification/api_v2_train.py @@ -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, -- GitLab