keras/keras_core
Haifeng Jin d579f815cd Add *_on_batch functions to torch trainer (#328)
* Add *_on_batch functions to torch trainer

* addressing comments

---------

Co-authored-by: Haifeng Jin <haifeng-jin@users.noreply.github.com>
2023-06-12 11:57:12 -07:00
..
activations Small ops fixes for Torch unit tests (#316) 2023-06-12 16:20:31 +00:00
applications Add on_batch methods in JAX and TF. 2023-06-07 18:26:42 -07:00
backend Add *_on_batch functions to torch trainer (#328) 2023-06-12 11:57:12 -07:00
callbacks Add error message for profiling in TB 2023-06-12 10:47:30 -07:00
constraints Small ops fixes for Torch unit tests (#316) 2023-06-12 16:20:31 +00:00
datasets Add WGAN example and fix minor issues. 2023-06-09 13:34:12 -07:00
initializers Add QR op and fix orthogonal initializer. 2023-06-10 11:58:13 -07:00
layers Fix Aurelien's new issue 2023-06-12 10:05:11 -07:00
losses Fix Aurelien's issue 2023-06-11 22:36:06 -07:00
metrics Add 2 timeseries examples 2023-06-10 12:24:24 -07:00
mixed_precision Refactor backend. 2023-05-03 15:33:40 -07:00
models Convert functional inputs to correct dtype (#326) 2023-06-12 11:47:24 -07:00
operations Small ops fixes for Torch unit tests (#316) 2023-06-12 16:20:31 +00:00
optimizers Extensive refactor of tracking logic. 2023-06-05 18:39:26 -07:00
random Fix dropout when noise_shape contains None (#264) 2023-06-05 16:40:19 +00:00
regularizers Add lowercase aliases for some regularizers for backwards compatibility. (#304) 2023-06-09 10:04:36 -07:00
saving Remove leftover statement 2023-06-12 09:40:18 -07:00
testing Merge branch 'main' of github.com:keras-team/keras-core 2023-06-04 21:48:30 -07:00
trainers Add *_on_batch functions to torch trainer (#328) 2023-06-12 11:57:12 -07:00
utils Minor summary formatting fix 2023-06-11 22:36:15 -07:00
__init__.py Touch ups 2023-05-13 17:33:48 -07:00
api_export.py Added initial Adam tests 2023-04-24 22:07:56 +00:00