From 37ddc8943b888c58e489f3f3e78c16ddf7e23c11 Mon Sep 17 00:00:00 2001
From: wangchaochaohu <wangchao66@baidu.com>
Date: Tue, 8 Oct 2019 14:42:08 +0800
Subject: [PATCH] refine build strategy for the model optimize test=develop
 (#3435)

* refine build strategy for the model optimize test=develop

* fix according develop test=develop
---
 tools/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tools/train.py b/tools/train.py
index 159cacc58..b9099210e 100644
--- a/tools/train.py
+++ b/tools/train.py
@@ -158,6 +158,7 @@ def main():
     # compile program for multi-devices
     build_strategy = fluid.BuildStrategy()
     build_strategy.fuse_all_optimizer_ops = False
+    build_strategy.fuse_elewise_add_act_ops = True
     # only enable sync_bn in multi GPU devices
     sync_bn = getattr(model.backbone, 'norm_type', None) == 'sync_bn'
     build_strategy.sync_batch_norm = sync_bn and devices_num > 1 \
-- 
GitLab