WaveGrad/configs/default.json at 8ae9d5312620e7a633bb9d2f5600992738468947 · ivanvovk/WaveGrad · 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
{
    "model_config": {
        "factors": [5, 5, 3, 2, 2],
        "upsampling_preconv_out_channels": 768,
        "upsampling_out_channels": [512, 512, 256, 128, 128],
        "upsampling_dilations": [
            [1, 2, 4, 8],
            [1, 2, 4, 8],
            [1, 2, 4, 8],
            [1, 2, 1, 2],
            [1, 2, 1, 2]
        ],
        "downsampling_preconv_out_channels": 32,
        "downsampling_out_channels": [128, 128, 256, 512],
        "downsampling_dilations": [
            [1, 2, 4], [1, 2, 4], [1, 2, 4], [1, 2, 4]
        ]
    },
    "data_config": {
        "sample_rate": 22050,
        "n_fft": 1024,
        "win_length": 1024,
        "hop_length": 300,
        "f_min": 80.0,
        "f_max": 8000,
        "n_mels": 80
    },
    "training_config": {
        "logdir": "logs/default",
        "continue_training": false,
        "train_filelist_path": "filelists/train.txt",
        "test_filelist_path": "filelists/test.txt",
        "batch_size": 96,
        "segment_length": 7200,
        "lr": 1e-3,
        "grad_clip_threshold": 1,
        "scheduler_step_size": 1,
        "scheduler_gamma": 0.9,
        "n_epoch": 100000000,
        "n_samples_to_test": 4,
        "test_interval": 1,
        "use_fp16": true,

        "training_noise_schedule": {
            "n_iter": 1000,
            "betas_range": [1e-6, 0.01]
        },
        "test_noise_schedule": {
            "n_iter": 50,
            "betas_range": [1e-6, 0.01]
        }
    },
    "dist_config": {
        "MASTER_ADDR": "localhost",
        "MASTER_PORT": "600010"
    }
}