未验证 提交 9dc7e5ef 编写于 作者: L Leo Guo 提交者: GitHub

Fix the parameter check error in rmsprop_kernel_xpu. (#52866)

上级 28de4558
......@@ -41,12 +41,6 @@ void RmspropDenseKernel(const Context& dev_ctx,
DenseTensor* mean_grad_out,
DenseTensor* master_param_outs) {
// copy learning_rate to cpu
PADDLE_ENFORCE_EQ(
learning_rate.dims().size(),
1,
errors::InvalidArgument("learining rate should have dimension = 1."
" But received learning rate dim [%s] ",
learning_rate.dims().size()));
T learning_rate_cpu = 0.0f;
memory_utils::Copy(CPUPlace(),
static_cast<void*>(&learning_rate_cpu),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册