keras/keras_core/trainers
Neel Kovelamudi 3e69e68338 Add adam optimizer and initial tests (#36)
* Adds adam optimizer

* Adds adam optimizer

* Add optimizer tests for Adam.

* Adds adam optimizer

* Added initial Adam tests

* Adds adam optimizer

* Applied nit fixes for Adam

* Indent fix in docstring
2023-04-25 01:46:03 +00:00
..
data_adapters Merge branch 'main' of github.com:keras-team/keras-core 2023-04-21 23:16:51 -07:00
__init__.py Start work on TF Trainer and DataAdapters. 2023-04-16 20:57:14 -07:00
compile_utils_test.py Misc fixes + add support for JAX array input 2023-04-17 17:47:22 -07:00
compile_utils.py Merge branch 'main' of github.com:keras-team/keras-core 2023-04-21 23:16:51 -07:00
epoch_iterator_test.py Add EpochIterator abstraction 2023-04-17 14:56:16 -07:00
epoch_iterator.py Switch TF trainer to a setup that enables future step fusing 2023-04-19 11:09:43 -07:00
trainer_test.py Merge branch 'main' of github.com:keras-team/keras-core 2023-04-21 23:16:51 -07:00
trainer.py Add adam optimizer and initial tests (#36) 2023-04-25 01:46:03 +00:00