-
Notifications
You must be signed in to change notification settings - Fork 7.4k
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
[GSoC] Autoencoder migration #68
base: master
Are you sure you want to change the base?
Conversation
Vishal-V
commented
Jun 12, 2019
- Migrated the autoencoder_runner.py and Autoencoder.py to Tensorflow 2.0
- The runner uses tf.data.Dataset pipeline to feed the batches and tf.keras.datasets to load the initial data and all necessary migrations in runner have been tested
- The backpropagation is in Eager mode with tf.GradientTape
- Wrote the Encoder and Decoder as subclasses of tf.keras.layers.Layer and the Autoencoder as a subclass of tf.keras.Model
- Tested the model with both Eager mode training as well as Graph mode training with model.fit.
This looks fine, but we might not be able to keep the stand alone copy of the code in tensorflow_examples/models, I'll check with the owners on the merge. |
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.
LGTM.
Thanks.
Thanks for the contribution but this |