tensorflow / tensor2tensor

Library of deep learning models and datasets designed to make deep learning more accessible and accelerate ML research.
Apache License 2.0
15.11k stars 3.44k forks source link

Migrating T2T to TF2 #1927

Open branko-fathom opened 1 year ago

branko-fathom commented 1 year ago

My org is looking to migrate to TensorFlow 2.7.4 by the end of June in order to not lose TPU access in GCP. The current plan for our fork is to utilize the tensorflow.compat.v1 APIs (along with updating contrib imports to use tensorflow_addons and tf_slim) and tf.disable_v2_behavior(). Is this all that's required to get t2t working with TF2?