Commit Graph

8 Commits (7ea6b1d1ca96f91210948ddfdd4c1dc11cc56187)

Author SHA1 Message Date
dangqingqing d432b10d8a Update cuda kernel and doc. 7 years ago
dangqingqing 5bd1e73f5e Refine and speedup momentum operator. 7 years ago
Siddharth Goyal 2e355f032e
Fix attribute naming for momentum_op () 7 years ago
kexinzhao cb0118f3e5 Polish Operator Doc (m) () 7 years ago
kavyasrinet 5380a5471b Adding Nesterov Momentum () 8 years ago
QI JUN 521514da36 fix conflict () 8 years ago
sidgoyal78 c10da26cf5 Modify implementation 8 years ago
sidgoyal78 d28b3094dd Add momentum operator 8 years ago