Change unittest's weight dimension to 1

9 years ago · ae57b4e774
parent a7730d409b
commit ae57b4e774
1 changed files with 1 additions and 1 deletions
--- 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,