fix ut test_recognize_digits (#24794)

paddle/fluid/train/test_train_recognize_digits.cc

@@ -19,6 +19,7 @@ limitations under the License. */
 #include "gtest/gtest.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/program_desc.h"
 #include "paddle/fluid/framework/tensor_util.h"

python/paddle/fluid/tests/book/test_recognize_digits.py

@@ -123,8 +123,8 @@ def train(nn_type,
                     # get test acc and loss
                     acc_val = numpy.array(acc_set).mean()
                     avg_loss_val = numpy.array(avg_loss_set).mean()
-                    if float(acc_val
-                             ) > 0.2:  # Smaller value to increase CI speed
+                    if float(acc_val) > 0.2 or pass_id == (PASS_NUM - 1):
+                        # Smaller value to increase CI speed
                         if save_dirname is not None:
                             fluid.io.save_inference_model(
                                 save_dirname, ["img"], [prediction],