This is a PyTorch implementation of Categorical Info WaveGAN (ciwGAN) and Featural Info WaveGAN (fiwGAN) from Beguš, 2021. The original code (in Tensorflow 1.12) can be found here.

python train.py --datadir training_directory --logdir log_directory

python train.py --ciw --num_categ N --datadir training_directory --logdir log_directory

python train.py --fiw --num_categ N --datadir training_directory --logdir log_directory

Add --cont last to the end of the training statement to continue from the last found state
Add --cont epoch_number to continue from the state corresponding to epoch_number