fix paddle/understand_sentiment_dynamic_lstm.py (the size of last bat…

…ch is smaller than batch_size )
dzhwinter · Dec 12, 2017 · b3584a7 · b3584a7
1 parent d3b8e07
commit b3584a7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/understand_sentiment_dynamic_lstm.py b/paddle/understand_sentiment_dynamic_lstm.py
@@ -128,7 +128,7 @@ def run_benchmark(model, args):
             tensor_words = to_lodtensor(map(lambda x: x[0], data), place)
 
             label = np.array(map(lambda x: x[1], data)).astype("int64")
-            label = label.reshape([args.batch_size, 1])
+            label = label.reshape([len(data), 1])
 
             tensor_label = fluid.LoDTensor()
             tensor_label.set(label, place)