Commit c03d5cc6 authored by Ali Siddiqui's avatar Ali Siddiqui Committed by Rasmus Munk Larsen
Browse files

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
parent 15d9f00f
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment