Add tf_keras imports to prepare for Keras 3 #28588
Rocketknight1
merged
19 commits into
main
from
import_tf_keras
amyeroberts
approved these changes
on 2024-01-29
Port core files + ESM (because ESM code is odd)
f0ddbd8e
Search-replace in modelling code
0de7d714
Fix up transfo_xl as well
728b7a18
Fix other core files + tests (still need to add correct import to tests)
316ea587
Fix cookiecutter
dbe1e059
make fixup, fix imports in some more core files
9201f68d
Auto-add imports to tests
3d86ddff
Cleanup, add imports to sagemaker tests
f0fd9bfe
Use correct exception for importing tf_keras
f4ba81e9
Fixes in modeling_tf_utils
cc6aaa08
make fixup
a2d325c8
Correct version parsing code
ea92cd3f
Ensure the pipeline tests correctly revert to float32 after each test
166b093c
Ensure the pipeline tests correctly revert to float32 after each test
d7bb7b63
More tf.keras -> keras
42242064
Add dtype cast
df5b4860
Better imports of tf_keras
193cbca5
Add a cast for tf.assign, just in case
383bc79e
Fix callback imports
b5e59ca3
Login to write a write a comment.
Login via GitHub