Commit Graph

7 Commits (2e101df7c656d524c92b6c31711a8bbcaf7ce09f)

Author SHA1 Message Date
dangqingqing e03b574e0e Refine the Eigen usage for CPU implementation.
7 years ago
dangqingqing 5bd1e73f5e Refine and speedup momentum operator.
7 years ago
Siddharth Goyal 2e355f032e
Fix attribute naming for momentum_op (#5453)
7 years ago
kavyasrinet 5380a5471b Adding Nesterov Momentum (#4948)
7 years ago
sidgoyal78 db77937ea4 Fix learning_rate usage for momentum
7 years ago
sidgoyal78 c10da26cf5 Modify implementation
7 years ago
sidgoyal78 d28b3094dd Add momentum operator
7 years ago