hparams.py

import tensorflow as tf

hparams = tf.contrib.training.HParams(
    
    # Audio:
    num_mcep=34,
    fs=22050,
    fftl=1024,
    shiftms=5.0,
    minf0=40.0,
    maxf0=500.0,
    
    # Model:
    num_features=64,
    num_classes=5,

    # Data loader
    num_workers=4,
    crop_size=(35, 128),
    source_limit=200,
    shuffle=True,

    # Training:
    batch_size=8,
    num_epochs = 1000,
    learning_rate_gen=2e-4,
    learning_rate_disc=1e-4,
    weight_decay=1e-5,
    betas=(0.5,0.999),

    # Save
    checkpoint_interval=10,

    # Eval:
    max_iters=200,
    griffin_lim_iters=60,
    power=1.5,
)

def hparams_debug_string():
    values = hparams.values()
    hp = ['  %s: %s' % (name, values[name]) for name in sorted(values)]
    return 'Hyperparameters:\n' + '\n'.join(hp)