diff --git a/tools/export_model.py b/tools/export_model.py index 01aba06c1f715f764352c6fd38a23c470e66e289..35f432f50e9d1dd903be3d0d3e07a4e42f2a2b7f 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()