modify light mnist file

bc8ba1de · zhuoyuan · fd5c7700 · bc8ba1de
显示空白变更内容
内联并排

Showing with 0 addition and 5 deletion

demo/mnist/light_mnist.py demo/mnist/light_mnist.py +0 -5

未找到文件。
--- 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)