From 226f775b3eec5382f8f1d540c579dba4f62727d5 Mon Sep 17 00:00:00 2001 From: Kaipeng Deng Date: Mon, 19 Aug 2019 11:44:35 +0800 Subject: [PATCH] fix enable_inplace (#3131) --- tools/infer.py | 2 +- tools/train.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/infer.py b/tools/infer.py index 769fda765..ac1fd215b 100644 --- a/tools/infer.py +++ b/tools/infer.py @@ -57,7 +57,7 @@ def get_save_image_name(output_dir, image_path): """ if not os.path.exists(output_dir): os.makedirs(output_dir) - image_name = image_path.split('/')[-1] + image_name = os.path.split(image_path)[-1] name, ext = os.path.splitext(image_name) return os.path.join(output_dir, "{}".format(name)) + ext diff --git a/tools/train.py b/tools/train.py index a4e6f40f2..619ea972b 100644 --- a/tools/train.py +++ b/tools/train.py @@ -134,7 +134,7 @@ def main(): # compile program for multi-devices build_strategy = fluid.BuildStrategy() build_strategy.memory_optimize = False - build_strategy.enable_inplace = True + build_strategy.enable_inplace = False sync_bn = getattr(model.backbone, 'norm_type', None) == 'sync_bn' # only enable sync_bn in multi GPU devices build_strategy.sync_batch_norm = sync_bn and devices_num > 1 \ -- GitLab