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