modify api_train_v2

8 years ago · 3b69629d32
parent 47043fe1d3
commit 3b69629d32
1 changed files with 2 additions and 2 deletions
--- 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)
-    label = paddle.layer.data(name='label', size=10)
+    images = paddle.layer.data(name='pixel', type=dense_vector(784))
+    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,