未验证 提交 ac347fce 编写于 作者: Z Zhen Wang 提交者: GitHub

add more error info for these ops without double grad ops. (#25989)

上级 50c68859
...@@ -887,7 +887,10 @@ void PartialGradTask::RunEachOp(OpBase *op) { ...@@ -887,7 +887,10 @@ void PartialGradTask::RunEachOp(OpBase *op) {
op->Attrs(), op->place()); op->Attrs(), op->place());
PADDLE_ENFORCE_NOT_NULL( PADDLE_ENFORCE_NOT_NULL(
double_grad_node, double_grad_node,
platform::errors::NotFound("The Op %s doesn't have any grad op.", platform::errors::NotFound("The Op %s doesn't have any grad op. If you "
"don't intend calculating higher order "
"derivatives, please set `create_graph` to "
"False.",
op->Type())); op->Type()));
VLOG(10) << "Create " << double_grad_node->size() VLOG(10) << "Create " << double_grad_node->size()
<< " double grad op(s) for " << op->Type() << " double grad op(s) for " << op->Type()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册