Skip to content

Commit 160641e

Browse files
Update python/paddle/optimizer/lbfgs.py
update warning description Co-authored-by: HydrogenSulfate <[email protected]>
1 parent b3d7006 commit 160641e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/optimizer/lbfgs.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def check_tf32_override():
6363
warnings.warn(
6464
"Warning! TF32 Tensor Cores are enabled by default on some NVIDIA GPUs for faster computation, "
6565
"but may compromise numerical precision in specific cases, particularly with the L-BFGS optimizer. "
66-
"To disable it, set os.environ['NVIDIA_TF32_OVERRIDE'] = '0'."
66+
"To disable it, set: NVIDIA_TF32_OVERRIDE=0"
6767
)
6868

6969

0 commit comments

Comments
 (0)