提交 ddc90686 编写于 作者: Z Zeng Jinle 提交者: GitHub

set enable_inplace to True in models, test=develop (#3297)

上级 f4edc029
...@@ -134,7 +134,6 @@ def main(): ...@@ -134,7 +134,6 @@ def main():
# compile program for multi-devices # compile program for multi-devices
build_strategy = fluid.BuildStrategy() build_strategy = fluid.BuildStrategy()
build_strategy.enable_inplace = False
sync_bn = getattr(model.backbone, 'norm_type', None) == 'sync_bn' sync_bn = getattr(model.backbone, 'norm_type', None) == 'sync_bn'
# only enable sync_bn in multi GPU devices # only enable sync_bn in multi GPU devices
build_strategy.sync_batch_norm = sync_bn and devices_num > 1 and cfg.use_gpu build_strategy.sync_batch_norm = sync_bn and devices_num > 1 and cfg.use_gpu
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册