提交 7ba55aa2 编写于 作者: X Xin Pan

fix CMAKE

上级 69742652
...@@ -25,4 +25,6 @@ function(train_test TARGET_NAME) ...@@ -25,4 +25,6 @@ function(train_test TARGET_NAME)
endfunction(train_test) endfunction(train_test)
train_test(recognize_digits ARGS mlp conv) if(WITH_TESTING)
train_test(recognize_digits ARGS mlp conv)
endif()
...@@ -52,8 +52,6 @@ void Train() { ...@@ -52,8 +52,6 @@ void Train() {
PADDLE_ENFORCE_NE(loss_name, "", "loss not found"); PADDLE_ENFORCE_NE(loss_name, "", "loss not found");
// init all parameters
// prepare data // prepare data
auto x_var = scope.Var("img"); auto x_var = scope.Var("img");
auto x_tensor = x_var->GetMutable<framework::LoDTensor>(); auto x_tensor = x_var->GetMutable<framework::LoDTensor>();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册