Implementing the Adamax optimizer operator (#4538)
* 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
Showing
paddle/operators/adamax_op.cc
0 → 100644
paddle/operators/adamax_op.cu
0 → 100644
paddle/operators/adamax_op.h
0 → 100644
想要评论请 注册 或 登录