config.json

{
    "train_config": {
        "fp16_run": true,
        "output_directory": "checkpoints",
        "epochs": 100000,
        "init_lr": 4e-4,
        "final_lr": 1e-5,
        "sigma": 1.0,
        "epochs_per_checkpoint": 1,
        "batch_size": 24,
        "seed": 1234,
        "checkpoint_path": "",
        "with_tensorboard": true
    },
    "data_config": {
        "data_path": "voc_mol",
        "segment_length": 16384,
        "sampling_rate": 22050,
        "filter_length": 1024,
        "hop_length": 256,
        "win_length": 1024,
        "mel_pad_val": -4.0
    },
    "dist_config": {
        "dist_backend": "nccl",
        "dist_url": "tcp://localhost:54321"
    },

    "waveglow_config": {
        "n_mel_channels": 80,
        "n_flows": 12,
        "n_group": 32,
        "n_early_every": 4,
        "n_early_size": 8,
        "n_classes": 512,
        "WN_config": {
            "n_layers": 8,
            "n_channels": 256,
            "kernel_size": 3
        }
    }
}