tensorflow / gan

Tooling for GANs in TensorFlow
Apache License 2.0
930 stars 247 forks source link

Update to use keras? #41

Open steventrouble opened 3 years ago

steventrouble commented 3 years ago

Hi, this library doesn't seem to have a keras.Model. Keras is the main way many of us interact with TF, so it would be great to see here.

The TF docs have an example GAN class that works decently well:

https://www.tensorflow.org/guide/keras/customizing_what_happens_in_fit#wrapping_up_an_end-to-end_gan_example

But there's no official TF implementation committed anywhere. Would it be possible to add an official TF-GAN keras Model like the one in that example?

Being able to use Keras makes it much easier for people to learn and experiment with ML, so it would be awesome to see one here.

Thanks!

ebursztein commented 3 years ago

We have an initial branch for TF2 but its not yet complete and there is no ETA on it. Agree it seems useful to have a Keras version.

GeorgiaA commented 2 years ago

Hi. Has there been any further update to an ETA on a TF2 branch of this library? Thanks.