Skip to content

Commit 7b55357

Browse files
committed
fix nvprof gpu consistence
1 parent ef0d654 commit 7b55357

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

paddle/understand_sentiment_dynamic_lstm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def run_benchmark(model, args):
128128
tensor_words = to_lodtensor(map(lambda x: x[0], data), place)
129129

130130
label = np.array(map(lambda x: x[1], data)).astype("int64")
131-
label = label.reshape([args.batch_size, 1])
131+
label = label.reshape([len(data), 1])
132132

133133
tensor_label = fluid.LoDTensor()
134134
tensor_label.set(label, place)

0 commit comments

Comments
 (0)