modify api_train_v2

3b69629d · qiaolongfei · 47043fe1 · 3b69629d
显示空白变更内容
内联并排

Showing with 2 addition and 2 deletion

demo/mnist/api_train_v2.py demo/mnist/api_train_v2.py +2 -2

未找到文件。
--- a/demo/mnist/api_train_v2.py
+++ b/demo/mnist/api_train_v2.py
@@ -16,8 +16,8 @@ def main():
    paddle.init(use_gpu=False, trainer_count=1)
    # define network topology
-    images = paddle.layer.data(name='pixel', size=784)
+    images = paddle.layer.data(name='pixel', type=dense_vector(784))
-    label = paddle.layer.data(name='label', size=10)
+    label = paddle.layer.data(name='label', type=integer_value(10))
    hidden1 = paddle.layer.fc(input=images, size=200)
    hidden2 = paddle.layer.fc(input=hidden1, size=200)
    inference = paddle.layer.fc(input=hidden2,