Vocoder/config.json at main · aizamaksutova/Vocoder · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
{
  "name": "train_config",
  "n_gpu": 1,
  "arch": {
    "type": "HiFiGAN",
    "args": {
      "generator_config": {
        "in_channels": 80,
        "hidden_dimension": 512,
        "conv_kernel_sz": [16, 16, 4, 4],
        "mrf_kernel_sz": [3, 7, 11],
        "mrf_dilations": [
          [[1, 1], [3, 1], [5, 1]],
          [[1, 1], [3, 1], [5, 1]],
          [[1, 1], [3, 1], [5, 1]]
        ]
      },
      "mpd_config": {
        "periods": [2, 3, 5, 7, 11],
        "kernel_size": 5,
        "stride": 3,
        "channels": [32, 128, 512, 1024]
      },
      "msd_config": {
        "factors": [1, 2, 4],
        "kernel_sizes": [15, 41, 41, 41, 41, 41, 5],
        "strides": [1, 2, 2, 4, 4, 1, 1],
        "groups": [1, 4, 16, 16, 16, 16, 1],
        "channels": [128, 128, 256, 512, 1024, 1024, 1024]
      }
    }
  },
  "data": {
    "train": {
      "batch_size": 7,
      "num_workers": 5,
      "datasets": [
        {
          "type": "LjDataset",
          "args": {
            "data_path": "data/LJSpeech-1.1/wavs",
            "wav_max_len": 22272
          }
        }
      ]
    }
  },
  "optimizer_d": {
    "type": "AdamW",
    "args": {
      "lr": 2e-4,
      "betas": [0.8, 0.99],
      "weight_decay": 0.01
    }
  },
  "optimizer_g": {
    "type": "AdamW",
    "args": {
      "lr": 2e-4,
      "betas": [0.8, 0.99],
      "weight_decay": 0.01
    }
  },
  "lr_scheduler_d": {
    "type": "ExponentialLR",
    "args": {
      "gamma": 0.9
    }
  },
  "lr_scheduler_g": {
    "type": "ExponentialLR",
    "args": {
      "gamma": 0.9
    }
  },
  "loss": {
    "type": "HiFiGANLoss",
    "args": {
      "mel_spectrogram_multiplier": 45,
      "feature_matching_multiplier": 2
    }
  },
  "metrics": [],
  "trainer": {
    "epochs": 10000,
    "save_dir": "saved/",
    "save_period": 10,
    "verbosity": 2,
    "monitor": "min val_loss",
    "early_stop": 500,
    "visualize": "wandb",
    "wandb_project": "vocoder",
    "wandb_run_name": "first real train run",
    "len_epoch": 1000,
    "grad_norm_clip": 1000
  }
}