NADAM Optimizer (#9889)
* Initial commit for NADAM * Add GPU and sparse implementations and add missing arguments * Add tester * Revert changes to files made for testing * Add nadam optimizer in a class of its own * Reverse changes to adam_test.py * Reverse changes to adam.py * Actually reverse adam_test.py * Actually reverse adam.py * Delete nadam_optimizer_test.py * Create nadam_optimizer_test.py * Fix BUILD * Run buildifier on BUILD
Loading
Please sign in to comment