提交 a14fcd0f 编写于 作者: C cuicheng01

update export_model.py

上级 810588ce
...@@ -30,5 +30,7 @@ if __name__ == "__main__": ...@@ -30,5 +30,7 @@ if __name__ == "__main__":
args = config.parse_args() args = config.parse_args()
config = config.get_config( config = config.get_config(
args.config, overrides=args.override, show=False) args.config, overrides=args.override, show=False)
if config["Arch"].get("use_sync_bn", False):
config["Arch"]["use_sync_bn"] = False
engine = Engine(config, mode="export") engine = Engine(config, mode="export")
engine.export() engine.export()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册