RawNet2/config.json at main · aizamaksutova/RawNet2 · 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
{
  "name": "train_config",
  "n_gpu": 1,
  "arch": {
      "type": "RawNet2",
      "args": {
          "sinc_channels": 128,
          "sinc_filter_length": 1024,
          "channels1": 20,
          "channels2": 128,
          "gru_hidden_size": 1024,
          "sinc_filter_type": "mel-scaled"
      }
  },
  "data": {
      "train": {
          "batch_size": 32,
          "num_workers": 1,
          "datasets": [
              {
                  "type": "ASVDataset",
                  "args": {
                      "dir": "/users/aizam/RawNet2/data",
                      "part": "train",
                      "max_sec_length": 4
                  }
              }
          ]
      },
      "test": {
          "batch_size": 64,
          "num_workers": 1,
          "datasets": [
              {
                  "type": "ASVDataset",
                  "args": {
                      "dir": "/users/aizam/RawNet2/data",
                      "part": "eval",
                      "max_sec_length": 4
                  }
              }
          ]
      }
  },
  "optimizer": {
      "type": "AdamW",
      "args": {
          "lr": 1e-4,
          "weight_decay": 1e-4
      }
  },
  "loss": {
    "type": "CrossEntropyLoss_RawNet",
    "args": {
      "weight": [1, 9]
    }
},
  "metrics":[
      {
          "type": "EER",
          "args": {}
      }
  ],
  "lr_scheduler": {
      "type": "ExponentialLR",
      "args": {
          "gamma": 0.99999
      }
  },
  "trainer": {
      "epochs": 50,
      "len_epoch": 4000,
      "log_step": 100,
      "save_dir": "saved/",
      "save_period": 1,
      "verbosity": 2,
      "monitor": "min val_loss",
      "early_stop": 100,
      "visualize": "wandb",
      "wandb_project": "RawNet2",
      "grad_norm_clip": 10
  }
}