Matt 349f1c85d3 Rewrite TensorFlow train_step and test_step (#17057) ...

* Initial commit

* Better label renaming

* Remove breakpoint before pushing (this is your job)

* Test a lot more in the Keras fit() test

* make fixup

* Clarify the case where we flatten y dicts into tensors

* Clarify the case where we flatten y dicts into tensors

* Extract label name remapping to a method

2022-05-17 14:36:23 +01:00

84 KiB

Raw Blame History

View Raw