py-func机制使用,报告错误
收藏
使用py_func机制,实现个类似交叉熵损失的层,
碰到个问题, 在自己反向梯度计算函数中, 我返回一个参数,运行的时候报错如下,
Error: Output number not match. Expected 2, actual 1 (/paddle/paddle/fluid/operators/py_func_op.cc:77)
如果我返回两个参数,就可以通过了, 反向梯度计算函数 不就是返回 误差参数吗,还需要返回什么参数
0
收藏
请登录后评论
PaddlePaddle 的优化器返回的是 tuple:
返回: (optimize_ops, params_grads),数据类型为(list, list),其中optimize_ops是minimize接口为网络添加的OP列表,params_grads是一个由(param, grad)变量对组成的列表,param是Parameter,grad是该Parameter对应的梯度值
返回类型: tuple
相关文档:
https://www.paddlepaddle.org.cn/documentation/docs/zh/api_guides/low_level/optimizer.html