未验证 提交 a125697e 编写于 作者: Q Qi Li 提交者: GitHub

fix ut test_recognize_digits (#24794)

上级 75eec3d1
...@@ -19,6 +19,7 @@ limitations under the License. */ ...@@ -19,6 +19,7 @@ limitations under the License. */
#include "gtest/gtest.h" #include "gtest/gtest.h"
#include "paddle/fluid/framework/executor.h" #include "paddle/fluid/framework/executor.h"
#include "paddle/fluid/framework/io/fs.h"
#include "paddle/fluid/framework/op_registry.h" #include "paddle/fluid/framework/op_registry.h"
#include "paddle/fluid/framework/program_desc.h" #include "paddle/fluid/framework/program_desc.h"
#include "paddle/fluid/framework/tensor_util.h" #include "paddle/fluid/framework/tensor_util.h"
......
...@@ -123,8 +123,8 @@ def train(nn_type, ...@@ -123,8 +123,8 @@ def train(nn_type,
# get test acc and loss # get test acc and loss
acc_val = numpy.array(acc_set).mean() acc_val = numpy.array(acc_set).mean()
avg_loss_val = numpy.array(avg_loss_set).mean() avg_loss_val = numpy.array(avg_loss_set).mean()
if float(acc_val if float(acc_val) > 0.2 or pass_id == (PASS_NUM - 1):
) > 0.2: # Smaller value to increase CI speed # Smaller value to increase CI speed
if save_dirname is not None: if save_dirname is not None:
fluid.io.save_inference_model( fluid.io.save_inference_model(
save_dirname, ["img"], [prediction], save_dirname, ["img"], [prediction],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册