From bc8ba1de750637773052892c5c008f30cd998844 Mon Sep 17 00:00:00 2001 From: zhuoyuan Date: Tue, 7 Feb 2017 16:02:35 -0800 Subject: [PATCH] modify light mnist file --- demo/mnist/light_mnist.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/demo/mnist/light_mnist.py b/demo/mnist/light_mnist.py index 271f73cd4d1..466cc2e2f54 100644 --- a/demo/mnist/light_mnist.py +++ b/demo/mnist/light_mnist.py @@ -53,12 +53,7 @@ def light_cnn(input_image, num_channels, num_classes): tmp = __light__(tmp, num_filter=128) tmp = __light__(tmp, num_filter=128, conv_filter_size=1) - #tmp = img_pool_layer(input=tmp, stride=2, pool_size=2, pool_type=MaxPooling()) - #tmp = dropout_layer(input=tmp, dropout_rate=0.5) tmp = fc_layer(input=tmp, size = num_classes, act=SoftmaxActivation()) - # tmp = fc_layer(input=tmp, size=512, layer_attr=ExtraAttr(drop_rate=0.5), act=LinearActivation()) - # tmp = batch_norm_layer(input=tmp, act=ReluActivation()) - # return fc_layer(input=tmp, size=num_classes, act=SoftmaxActivation()) return tmp predict = light_cnn(input_image=img, num_channels=1, num_classes=label_size) -- GitLab