metadata

title: CycleGAN Flax
emoji: 📚
colorFrom: green
colorTo: yellow
sdk: docker
pinned: false

CycleGAN in Jax + Flax

This space showcases CycleGAN models that were trained on the horse2zebra and monet2photo TensorFlow datasets in Jax with Flax. Please see the Colab notebook for an example of how to instantiate the models and load the trained weights. It includes a Gradio demo that you can use to make your own demo. The model weights are also available on Hugging Face at : louwjac/CycleGAN-Flax.

TensorBoard logs

The TensorBoard logs for the training runs can be viewed at the links below. The logged images are unfortunately not available due to file size constraints.

horse2zebra
monet2photo