CycleGAN in Jax + Flax

This repository contains weights for 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 makes it easy to submit external images to the model. You can also test the models in the Spaces demo without running any code yourself.

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.

  1. horse2zebra
  2. monet2photo
Open in Spaces Open In Colab Custom badge
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.