Skip to content
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

Open
wants to merge 9 commits into
base: master
Choose a base branch
from

Conversation

Vishal-V
Copy link
Contributor

  • 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.

@googlebot googlebot added the cla: yes CLA has been signed label Jun 12, 2019
@MarkDaoust
Copy link
Member

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.

Copy link
Member

@MarkDaoust MarkDaoust left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

Thanks.

@lamberta
Copy link
Member

Thanks for the contribution but this community/ directory has been deprecated. We're no longer accepting pull requests here.
Not sure where the rest of this PR stands, though.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
cla: yes CLA has been signed ready to pull
Projects
None yet
4 participants