diff --git a/tools/infer.py b/tools/infer.py index 769fda7653c189474b9b94eeada8c1ed101d6c80..ac1fd215b9f540c8d78959864a041f43376e1be1 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 a4e6f40f2703b290c70fd873f19bab8df9570307..619ea972b0611316aa4b840491eb40d7f2056048 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 \