update export_model.py

pull/2042/head
cuicheng01 2022-06-13 10:38:29 +00:00
parent 810588ce1c
commit a14fcd0fb3
1 changed files with 2 additions and 0 deletions

View File

@ -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()