From 7e710c1c1c7b17bf2a276f875631ec4e886a08ca Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Fri, 18 Dec 2020 13:25:52 +0800
Subject: [PATCH] [Dygraph]set norm_type to bn while export model (#1922)

* set norm_type to bn while export model

* modify code according to review
---
 tools/export_model.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tools/export_model.py b/tools/export_model.py
index 8001e0fb1..26c0871db 100644
--- a/tools/export_model.py
+++ b/tools/export_model.py
@@ -101,6 +101,9 @@ def main():
     FLAGS = parse_args()
 
     cfg = load_config(FLAGS.config)
+    # TODO: to be refined in the future
+    if cfg.norm_type == 'sync_bn':
+        FLAGS.opt['norm_type'] = 'bn'
     merge_config(FLAGS.opt)
     check_config(cfg)
     check_gpu(cfg.use_gpu)
-- 
GitLab