未验证 提交 0f5e0ba1 编写于 作者: C chentianyu03 提交者: GitHub

【cherry-pick】add more info to tensor.grad warning message (#34264) #34288

Add more information to tensor.grad warning message.
上级 4ffd3395
...@@ -371,7 +371,10 @@ def monkey_patch_varbase(): ...@@ -371,7 +371,10 @@ def monkey_patch_varbase():
# Tensor(shape=[1], dtype=float32, place=CUDAPlace(0), stop_gradient=False, [500.]) # Tensor(shape=[1], dtype=float32, place=CUDAPlace(0), stop_gradient=False, [500.])
""" """
msg = "tensor.grad will return the tensor value of the gradient." msg = 'tensor.grad will return the tensor value of the gradient.' \
' This is an incompatible upgrade for tensor.grad API. ' \
' It\'s return type changes from numpy.ndarray in version 2.0 to paddle.Tensor in version 2.1.0. ' \
' If you want to get the numpy value of the gradient, you can use :code:`x.grad.numpy()`'
warning_msg = "\033[93m\nWarning:\n%s \033[0m" % (msg) warning_msg = "\033[93m\nWarning:\n%s \033[0m" % (msg)
# ensure ANSI escape sequences print correctly in cmd and powershell # ensure ANSI escape sequences print correctly in cmd and powershell
if sys.platform.lower() == 'win32': if sys.platform.lower() == 'win32':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册