“848ae7dc34c84c09ac6df93e5cfd5c2031156cea”上不存在“paddle/phi/kernels/mean_all_grad_kernel.h”
-
由 Abhinav Arora 提交于
* Implementing the Adamax optimizer step operator * Adding unit tests for adamax_op * Changing learning rate and time step to inputs from attributes * Changing learning rate and time step to input(tensors) * Making the Adamax operator conform to naming convention * Removing Tensor<float> from comments * Rectifying the Adamax implementation * Changing Unit Test values and adding comments * Changing Unit Test to test multiple steps
4cb5bd90