params.py

# DATA
SAMPLE_RATE = 16000
SLICE_LEN = 65536

RAW_DATA_DIR = "raw_data/"

PREPROCESSED_DATA_DIR = "preprocessed_data/"

# MODELS
NOISE_DIM = 100
WAVEGAN = "WaveGAN"
TRANSGAN = "TransGAN"

# TRAINING
BATCH_SIZE = 4
EPOCHS = 20
EPOCHS_PER_SAVE = 10
N_CRITIC = 5

LR_G = 1e-5
LR_D = 1e-5
BETA1 = 0.5
BETA2 = 0.9

PENALTY_COEFF = 10

MODEL_OUTPUT_DIR = "outputs/"