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
..
common	Minor fixes	2023-06-12 09:39:43 -07:00
jax	Fix Aurelien's new issue	2023-06-12 10:40:40 -07:00
tensorflow	Add backwards compat shim for model.loss in TF.	2023-06-10 11:58:32 -07:00
tests	Add compute output spec test	2023-05-20 20:38:21 -07:00
torch	Add *_on_batch functions to torch trainer (#328)	2023-06-12 11:57:12 -07:00
__init__.py	Rename pytorch to torch.	2023-05-15 13:56:52 -07:00
config.py	Merge branch 'main' of github.com:keras-team/keras-core	2023-06-05 13:44:43 -07:00