“142e832d21715c0ce651e4ac04f10554945e5ad7”上不存在“paddle/phi/kernels/prelu_kernel.h”
未验证 提交 3fda9376 编写于 作者: C cuicheng01 提交者: GitHub

Merge pull request #2042 from cuicheng01/update_export_model

update  export_model.py
...@@ -30,5 +30,7 @@ if __name__ == "__main__": ...@@ -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()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册