From a14fcd0fb3698517c192c4a461e36272f6623657 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Mon, 13 Jun 2022 10:38:29 +0000 Subject: [PATCH] update export_model.py --- tools/export_model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tools/export_model.py b/tools/export_model.py index 01aba06c1..35f432f50 100644 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -30,5 +30,7 @@ if __name__ == "__main__": args = config.parse_args() config = config.get_config( 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.export()