From ae57b4e774c4af14a2d996a59d429b883ee6b544 Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Mon, 20 Mar 2017 14:41:17 +0800 Subject: [PATCH] Change unittest's weight dimension to 1 --- python/paddle/v2/tests/test_layer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/v2/tests/test_layer.py b/python/paddle/v2/tests/test_layer.py index 5ccd3d6913e..dad4a5d9493 100644 --- a/python/paddle/v2/tests/test_layer.py +++ b/python/paddle/v2/tests/test_layer.py @@ -22,7 +22,7 @@ import paddle.v2.networks as networks pixel = layer.data(name='pixel', type=data_type.dense_vector(128)) label = layer.data(name='label', type=data_type.integer_value(10)) -weight = layer.data(name='weight', type=data_type.dense_vector(10)) +weight = layer.data(name='weight', type=data_type.dense_vector(1)) score = layer.data(name='score', type=data_type.dense_vector(1)) hidden = layer.fc(input=pixel, -- GitLab