• A
    Implementing the Adamax optimizer operator (#4538) · 4cb5bd90
    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
adamax_op.cu 825 字节