未验证 提交 7e710c1c 编写于 作者: W wangxinxin08 提交者: GitHub

[Dygraph]set norm_type to bn while export model (#1922)

* set norm_type to bn while export model

* modify code according to review
上级 17bbda7f
...@@ -101,6 +101,9 @@ def main(): ...@@ -101,6 +101,9 @@ def main():
FLAGS = parse_args() FLAGS = parse_args()
cfg = load_config(FLAGS.config) 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) merge_config(FLAGS.opt)
check_config(cfg) check_config(cfg)
check_gpu(cfg.use_gpu) check_gpu(cfg.use_gpu)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册