mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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
parent 1347619f78 a14fcd0fb3
commit 3fda93765d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tools/export_model.py
View File

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