-
Notifications
You must be signed in to change notification settings - Fork 1.6k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Migrate hparams plugin to Keras 3 and remove Keras 2 from requirements.txt #6759
Conversation
requirements.txt.
963682e
to
eee218f
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you!!
@@ -46,12 +38,12 @@ def _initialize_model(self, writer): | |||
"optimizer": "adam", | |||
HP_DENSE_NEURONS: 8, | |||
} | |||
self.model = keras.models.Sequential( | |||
self.model = tf.keras.models.Sequential( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just curious but I haven't tested this myself - does this need to be changed to tf.keras.Sequential
: https://keras.io/api/models/sequential/
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looking at the source tf.keras.model.Sequential and tf.keras.Sequential appear to point to the same thing.
…s.txt (tensorflow#6759) We must still install keras 2 as part of CI in order to successfully run graph plugin tests.
…s.txt (#6759) We must still install keras 2 as part of CI in order to successfully run graph plugin tests.
#6759 modified _keras_test.py to work with Keras 3 but it did not import correctly into the internal repository. It was therefore reverted in #6767. This changes the test to work both in the oss and internal repos. Googlers, see cl/612511563 for sample import results.
tensorflow#6759 modified _keras_test.py to work with Keras 3 but it did not import correctly into the internal repository. It was therefore reverted in tensorflow#6767. This changes the test to work both in the oss and internal repos. Googlers, see cl/612511563 for sample import results.
We must still install keras 2 as part of CI in order to successfully run graph plugin tests.