Skip to content

Tutorial code for training the wasserstein conditional Generative Adversarial Network (cGAN) with Gradient Penalty (i.e., WGANGP)

Notifications You must be signed in to change notification settings

icpac-igad/cGAN_tutorial

 
 

Repository files navigation

Overview of the cGAN code

There are three main parts to getting the cGAN up and running for regional post-processing of global Numerical Weather Prediction (NWP) forecasts. For simplicity these are modularised into sub-directories with individual instructions contained within. When training and running the cGAN, we recommend visiting and following instructions from the sub-directories in the following order:

  1. data: Loading data and creating tfrecords for training.
  2. model: Setting up the model architecture and training the model.
  3. scripts: Generating forecasts.

Additionally, sub-directories evaluation and config contain evaluation scripts and the necessary configuration files for setting data paths and model architecture.

About

Tutorial code for training the wasserstein conditional Generative Adversarial Network (cGAN) with Gradient Penalty (i.e., WGANGP)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 88.5%
  • Python 11.0%
  • HCL 0.5%