提交 192c49cb 编写于 作者: S sneaxiy

modification

上级 03a99a8a
...@@ -330,8 +330,6 @@ class AdamOpKernel : public framework::OpKernel<T> { ...@@ -330,8 +330,6 @@ class AdamOpKernel : public framework::OpKernel<T> {
merge_func(ctx.template device_context<DeviceContext>(), grad, merge_func(ctx.template device_context<DeviceContext>(), grad,
grad_merge_var); grad_merge_var);
grad_merge_ptr = grad_merge_var; grad_merge_ptr = grad_merge_var;
std::cerr << "Create new variables in adam_op" << std::endl;
} }
auto& grad_merge = *grad_merge_ptr; auto& grad_merge = *grad_merge_ptr;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册