mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

remove max inplace grad add (#3346)

pull/3349/head
hong 2025-01-26 17:19:10 +08:00 committed by GitHub
parent 5580de6077
commit bde3844bc4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppcls/engine/engine.py
View File

@ -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

1
ppcls/static/train.py
View File

@ -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)