Francois Chollet df6990f41a Misc fixes + add support for JAX array input		2023-04-17 17:47:22 -07:00
..
data_adapters	Misc fixes + add support for JAX array input	2023-04-17 17:47:22 -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	Add main loss tracking in TF trainer	2023-04-17 17:23:53 -07:00
epoch_iterator_test.py	Add EpochIterator abstraction	2023-04-17 14:56:16 -07:00
epoch_iterator.py	Misc fixes.	2023-04-17 15:41:48 -07:00
trainer.py	Add main loss tracking in TF trainer	2023-04-17 17:23:53 -07:00