wavenet_config.json

{
  "name": "wavenet-vocoder",
  "n_gpu": 2,
  "arch": {
    "type": "WaveNet",
    "args": {
      "n_blocks": 2,
      "n_layers": 10,
      "classes": 256,
      "radix": 2,
      "descending": false,
      "aux_channels": 80,
      "dilation_channels": 128,
      "residual_channels": 256,
      "skip_channels": 256
    }
  },
  "data_loader": {
    "type": "RandomWaveFileLoader",
    "args": {
      "data_dir": "data/",
      "batch_size": 10,
      "num_workers": 2,
      "segment": 5000,
      "quantization_channels": 256,
      "injected_noise": false
    }
  },
  "optimizer": {
    "type": "Adam",
    "args": {
      "lr": 2e-4
    }
  },
  "loss": "cross_entropy",
  "lr_scheduler": {
    "type": "StepLR",
    "args": {
      "step_size": 100000,
      "gamma": 0.5
    }
  },
  "trainer": {
    "steps": 500000,
    "save_dir": "saved/",
    "save_freq": 50000,
    "verbosity": 2
  },
  "visualization": {
    "tensorboardX": true,
    "log_dir": "saved/runs"
  },
  "feature": {
    "type": "get_logmel",
    "args": {
      "sr": 22050,
      "n_fft": 1024,
      "hop_length": 128,
      "n_mels":80,
      "fmax": 8000
    }
  }
}