{
    "train_config": {
        "fp16_run": true,
        "output_directory": "checkpoints",
        "epochs": 100000,
        "learning_rate": 4e-4,
        "sigma": 1.0,
        "iters_per_checkpoint": 2000,
        "batch_size": 96, 
        "seed": 1234,
        "checkpoint_path": "checkpoints/Squeeze_244000",
        "with_tensorboard": true
    },
    "data_config": {
        "training_files": "train_files.txt",
        "segment_length": 16384,
        "sampling_rate": 22050,
        "filter_length": 1024,
        "hop_length": 256,
        "win_length": 1024,
        "mel_fmin": 0.0,
        "mel_fmax": 8000.0
    },
    "dist_config": {
        "dist_backend": "nccl",
        "dist_url": "tcp://localhost:54321"
    },
    "squeezewave_config": {
        "n_mel_channels": 80,
        "n_flows": 12,
        "n_audio_channel": 128,
        "n_early_every": 2,
        "n_early_size": 16,
        "WN_config": {
            "n_layers": 8,
            "n_channels": 256,
            "kernel_size": 3
        }
    }
}
