File tree Expand file tree Collapse file tree 2 files changed +14
-9
lines changed
paddle/capi/examples/model_inference/dense Expand file tree Collapse file tree 2 files changed +14
-9
lines changed Original file line number Diff line number Diff line change @@ -2,5 +2,5 @@ project(dense)
2
2
cmake_minimum_required (VERSION 2.8 )
3
3
aux_source_directory (. SRC_LIST )
4
4
add_executable (${PROJECT_NAME} ${SRC_LIST} )
5
- set_property ( TARGET ${PROJECT_NAME} PROPERTY C_STANDARD 99 )
5
+ set ( CMAKE_CXX_FLAGS " ${CMAKE_CXX_FLAGS} -std=c++11" )
6
6
target_link_libraries (${PROJECT_NAME} -lpaddle_capi_shared )
Original file line number Diff line number Diff line change 1
1
#include < paddle/capi.h>
2
2
#include < time.h>
3
+ #include < iostream>
4
+ #include < vector>
3
5
#include " ../common/common.h"
4
6
5
7
#define CONFIG_BIN " ./trainer_config.bin"
6
8
7
9
int main () {
8
10
// Initalize Paddle
9
- char * argv [] = {"--use_gpu=False" };
10
- CHECK (paddle_init (1 , (char * * )argv ));
11
+ std::string comand [] = {" --use_gpu=False" };
12
+ CHECK (paddle_init (1 , (char **)comand ));
11
13
12
14
// Reading config binary file. It is generated by `convert_protobin.sh`
13
15
long size;
@@ -53,17 +55,20 @@ int main() {
53
55
54
56
CHECK (paddle_arguments_get_value (out_args, 0 , prob));
55
57
56
- std ::std :: vector < paddle_real > result ;
57
- int height ;
58
- int width ;
58
+ std::vector<paddle_real> result;
59
+ uint64_t height;
60
+ uint64_t width;
59
61
60
- CHECK (paddle_matrix_get_shape (prob , & height , & width );
62
+ CHECK (paddle_matrix_get_shape (prob, &height, &width)) ;
61
63
result.resize (height * width);
62
64
CHECK (paddle_matrix_get_value (prob, result.data ()));
63
65
64
- printf ("Prob: " );
66
+ printf (" Prob: \n " );
65
67
for (int i = 0 ; i < height * width; ++i) {
66
- printf ("%.2f " , result [i ]);
68
+ printf (" %.4f " , result[i]);
69
+ if ((i + 1 ) % width == 0 ){
70
+ printf (" \n " );
71
+ }
67
72
}
68
73
printf (" \n " );
69
74
You can’t perform that action at this time.
0 commit comments