diff --git a/fluid/face_detection/reader.py b/fluid/face_detection/reader.py index ae90ad6b51db274446785dd6d9def57a606747d2..ea0f293dbfae57920fb9fdd2427ff273b90f8db9 100644 --- a/fluid/face_detection/reader.py +++ b/fluid/face_detection/reader.py @@ -285,7 +285,8 @@ def train(settings, try: enqueuer = GeneratorEnqueuer( train_generator(settings, file_list, batch_size, shuffle), - use_multiprocessing=use_multiprocessing) + use_multiprocessing=use_multiprocessing, + wait_time=0.5) enqueuer.start(max_queue_size=max_queue, workers=num_workers) generator_output = None while True: @@ -294,7 +295,7 @@ def train(settings, generator_output = enqueuer.queue.get() break else: - time.sleep(0.02) + time.sleep(0.5) yield generator_output generator_output = None finally: diff --git a/fluid/face_detection/train.py b/fluid/face_detection/train.py index f083f1ffbc05b3ac384c1df10d5176b93cb4be8c..67cec03b95ba5ffe1a5230c287bd12a49b90bb34 100644 --- a/fluid/face_detection/train.py +++ b/fluid/face_detection/train.py @@ -167,7 +167,7 @@ def train(args, config, train_params, train_file_list): shutil.rmtree(model_path) print('save models to %s' % (model_path)) - fluid.io.save_persistables(exe, model_path) + fluid.io.save_persistables(exe, model_path, main_program=program) train_py_reader.start() try: