Skip to content

Commit ae57b4e

Browse files
committed
Change unittest's weight dimension to 1
1 parent a7730d4 commit ae57b4e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/v2/tests/test_layer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323
pixel = layer.data(name='pixel', type=data_type.dense_vector(128))
2424
label = layer.data(name='label', type=data_type.integer_value(10))
25-
weight = layer.data(name='weight', type=data_type.dense_vector(10))
25+
weight = layer.data(name='weight', type=data_type.dense_vector(1))
2626
score = layer.data(name='score', type=data_type.dense_vector(1))
2727

2828
hidden = layer.fc(input=pixel,

0 commit comments

Comments
 (0)