WGAN-GP β CIFAR-10 Airplanes
This repository contains a Wasserstein GAN with Gradient Penalty (WGAN-GP)
trained on CIFAR-10 airplane images.
Files
wgan_generator.kerasβ Trained generator (use for inference)wgan_critic.kerasβ Trained critictraining_config.jsonβ Training hyperparameters
Usage
import tensorflow as tf
from tensorflow import keras
generator = keras.models.load_model("wgan_generator.keras")
LATENT_DIM = 128
noise = tf.random.normal([9, LATENT_DIM])
images = (generator(noise, training=False) + 1) / 2
Inference Providers NEW
This model isn't deployed by any Inference Provider. π Ask for provider support