Merge pull request #2042 from cuicheng01/update_export_model

update  export_model.py
pull/2044/head
cuicheng01 2022-06-13 18:40:31 +08:00 committed by GitHub
commit 3fda93765d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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()