WaveSplit/config_WS.json at master · Sinica-SLAM/WaveSplit · 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
{
    "train_config": {
        "output_directory": "checkpoints_Wavesplit_reg_abs",
        "trainer_type": "basic",
        "epoch": 400,
        "optimize_param": {
            "optim_type": "Adam",
            "learning_rate": 2e-3,
            "max_grad_norm": 10,
            "lr_scheduler":{
                "step_size": 100000,
                "gamma": 0.5,
                "last_epoch": -1
            }
        },
        "batch_size": 8,
        "epochs_per_checkpoint": 5,
        "epochs_per_log": 1,
        "seed": 1234,
        "checkpoint_path": "",
        "num_workers": 4,
        "half_lr": "yes",
        "early_stop": "yes"
    },
    "test_config": {
        "checkpoint_directory": "checkpoints_Wavesplit_new_reg3",
        "batch_size": 8,
        "checkpoint_path": "best.pth",
        "num_workers": 4,
        "output_directory": "WaveSplit_Regularize_SNR"
    },
    "filterbank": {
        "n_filters": 256,
        "kernel_size": 32,
        "stride": 16,
        "padding": 16
    },
    "speakerstack": {
        "n_blocks_spk": 14,
        "n_repeats_spk": 1,
        "mask_act_spk": "relu",
        "bn_chan_spk": 128,
        "skip_chan_spk": 128,
        "hid_chan_spk": 256
    },
    "separationstack": {
        "n_blocks_sep": 11,
        "n_repeats_sep": 4,
        "mask_act_sep": "relu",
        "bn_chan_sep": 128,
        "skip_chan_sep": 128,
        "hid_chan_sep": 256
    },
    "data": {
        "train_dir": "wham_scripts/json_wavesplit/wav8k/min/tr/",
        "valid_dir": "wham_scripts/json_wavesplit/wav8k/min/cv/",
        "test_dir": "wham_scripts/json_wavesplit/wav8k/min/tt/",
        "task": "sep_clean",
        "nondefault_nsrc":"",
        "sample_rate": 8000,
        "mode": "min"
    }
}