fix sync_bn in cpu (PaddlePaddle#2825)

* fix sync_bn in cpu
heavengate · Apr 29, 2021 · 03f79b0 · 03f79b0
1 parent a73970b
commit 03f79b0
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/tools/eval.py b/tools/eval.py
@@ -117,6 +117,9 @@ def main():
 
     place = paddle.set_device('gpu' if cfg.use_gpu else 'cpu')
 
+    if 'norm_type' in cfg and cfg['norm_type'] == 'sync_bn' and not cfg.use_gpu:
+        cfg['norm_type'] = 'bn'
+
     if FLAGS.slim_config:
         cfg = build_slim_model(cfg, FLAGS.slim_config, mode='eval')
 

diff --git a/tools/infer.py b/tools/infer.py
@@ -141,6 +141,9 @@ def main():
 
     place = paddle.set_device('gpu' if cfg.use_gpu else 'cpu')
 
+    if 'norm_type' in cfg and cfg['norm_type'] == 'sync_bn' and not cfg.use_gpu:
+        cfg['norm_type'] = 'bn'
+
     if FLAGS.slim_config:
         cfg = build_slim_model(cfg, FLAGS.slim_config, mode='test')
 

diff --git a/tools/train.py b/tools/train.py
@@ -123,6 +123,9 @@ def main():
 
     place = paddle.set_device('gpu' if cfg.use_gpu else 'cpu')
 
+    if 'norm_type' in cfg and cfg['norm_type'] == 'sync_bn' and not cfg.use_gpu:
+        cfg['norm_type'] = 'bn'
+
     if FLAGS.slim_config:
         cfg = build_slim_model(cfg, FLAGS.slim_config)