From 4ea244b2320d39b883c3b64c62bc2d04c08fce6d Mon Sep 17 00:00:00 2001
From: wangchaochaohu <wangchao66@baidu.com>
Date: Thu, 27 Jun 2019 03:27:09 -0500
Subject: [PATCH] rcnn config (#2569)

---
 PaddleCV/rcnn/train.py | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/PaddleCV/rcnn/train.py b/PaddleCV/rcnn/train.py
index 34df6211..e5755bdd 100644
--- a/PaddleCV/rcnn/train.py
+++ b/PaddleCV/rcnn/train.py
@@ -91,8 +91,8 @@ def train():
     optimizer.minimize(loss)
     fetch_list = fetch_list + [lr]
 
-    fluid.memory_optimize(
-        fluid.default_main_program(), skip_opt_set=set(fetch_list))
+    for var in fetch_list:
+        var.persistable = True
 
     place = fluid.CUDAPlace(0) if cfg.use_gpu else fluid.CPUPlace()
     exe = fluid.Executor(place)
@@ -106,8 +106,19 @@ def train():
         fluid.io.load_vars(exe, cfg.pretrained_model, predicate=if_exist)
 
     if cfg.parallel:
-        train_exe = fluid.ParallelExecutor(
-            use_cuda=bool(cfg.use_gpu), loss_name=loss.name)
+        build_strategy = fluid.BuildStrategy()
+        build_strategy.memory_optimize = False
+        build_strategy.enable_inplace = True
+
+        exec_strategy = fluid.ExecutionStrategy()
+        exec_strategy.use_experimental_executor = True
+        exec_strategy.num_iteration_per_drop_scope = 10
+        train_exe = fluid.ParallelExecutor(use_cuda=bool(cfg.use_gpu), 
+                            loss_name=loss.name, 
+                            build_strategy=build_strategy, 
+                            exec_strategy=exec_strategy)
+    else:
+        train_exe = exe
 
     shuffle = True
     if cfg.enable_ce:
-- 
GitLab