Skip to content

Commit d6bd5b1

Browse files
bug fix in dense
1 parent 5d3e816 commit d6bd5b1

File tree

1 file changed

+3
-2
lines changed
  • paddle/capi/examples/model_inference/dense

1 file changed

+3
-2
lines changed

paddle/capi/examples/model_inference/dense/main.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
#include <time.h>
33
#include <iostream>
44
#include <vector>
5+
56
#include "../common/common.h"
67

78
#define CONFIG_BIN "./trainer_config.bin"
@@ -40,7 +41,7 @@ int main() {
4041
for (int i = 0; i < input.size(); ++i) {
4142
input[i] = rand() / ((float)RAND_MAX);
4243
}
43-
44+
4445
// Set value for the input matrix
4546
CHECK(paddle_matrix_set_value(mat, input.data()));
4647

@@ -66,7 +67,7 @@ int main() {
6667
printf("Prob: \n");
6768
for (int i = 0; i < height * width; ++i) {
6869
printf("%.4f ", result[i]);
69-
if ((i + 1) % width == 0){
70+
if ((i + 1) % width == 0) {
7071
printf("\n");
7172
}
7273
}

0 commit comments

Comments
 (0)