diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py index 311692f8c..c92818d60 100644 --- a/ppcls/engine/engine.py +++ b/ppcls/engine/engine.py @@ -645,7 +645,7 @@ class Engine(object): self.auto_cast = AutoCast(use_amp) self.scaler = build_scaler(use_amp) else: - AMP_RELATED_FLAGS_SETTING = {'FLAGS_max_inplace_grad_add': 8, } + AMP_RELATED_FLAGS_SETTING = {} if paddle.is_compiled_with_cuda(): AMP_RELATED_FLAGS_SETTING.update({ 'FLAGS_cudnn_batchnorm_spatial_persistent': 1 diff --git a/ppcls/static/train.py b/ppcls/static/train.py index a7a5c4af5..0ad138a96 100755 --- a/ppcls/static/train.py +++ b/ppcls/static/train.py @@ -104,7 +104,6 @@ def main(args): 'FLAGS_cudnn_exhaustive_search': 1, 'FLAGS_conv_workspace_size_limit': 1500, 'FLAGS_cudnn_batchnorm_spatial_persistent': 1, - 'FLAGS_max_inplace_grad_add': 8, } os.environ['FLAGS_cudnn_batchnorm_spatial_persistent'] = '1' paddle.set_flags(AMP_RELATED_FLAGS_SETTING)