mirrors/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-06-03 21:53:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove max inplace grad (#14596)

Browse Source 
* remove max inplace grad

* update
This commit is contained in:
hong 2025-01-26 17:19:42 +08:00 committed by GitHub
parent cf03afb10a
commit b18b656633
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
test_tipc/supplementary/train.py
View File

@ -66,7 +66,6 @@ def amp_scaler(config):
if "AMP" in config and config["AMP"]["use_amp"] is True: if "AMP" in config and config["AMP"]["use_amp"] is True:
AMP_RELATED_FLAGS_SETTING = { AMP_RELATED_FLAGS_SETTING = {
"FLAGS_cudnn_batchnorm_spatial_persistent": 1, "FLAGS_cudnn_batchnorm_spatial_persistent": 1,
"FLAGS_max_inplace_grad_add": 8,
} }
paddle.set_flags(AMP_RELATED_FLAGS_SETTING) paddle.set_flags(AMP_RELATED_FLAGS_SETTING)
scale_loss = config["AMP"].get("scale_loss", 1.0) scale_loss = config["AMP"].get("scale_loss", 1.0)

1
tools/eval.py
View File

@ -131,7 +131,6 @@ def main():
if use_amp: if use_amp:
AMP_RELATED_FLAGS_SETTING = { AMP_RELATED_FLAGS_SETTING = {
"FLAGS_cudnn_batchnorm_spatial_persistent": 1, "FLAGS_cudnn_batchnorm_spatial_persistent": 1,
"FLAGS_max_inplace_grad_add": 8,
} }
paddle.set_flags(AMP_RELATED_FLAGS_SETTING) paddle.set_flags(AMP_RELATED_FLAGS_SETTING)
scale_loss = config["Global"].get("scale_loss", 1.0) scale_loss = config["Global"].get("scale_loss", 1.0)

4
tools/train.py
View File

@ -181,9 +181,7 @@ def main(config, device, logger, vdl_writer, seed):
except: except:
pass pass
if use_amp: if use_amp:
AMP_RELATED_FLAGS_SETTING = { AMP_RELATED_FLAGS_SETTING = {}
"FLAGS_max_inplace_grad_add": 8,
}
if paddle.is_compiled_with_cuda(): if paddle.is_compiled_with_cuda():
AMP_RELATED_FLAGS_SETTING.update( AMP_RELATED_FLAGS_SETTING.update(
{ {