-
Notifications
You must be signed in to change notification settings - Fork 1
/
model_config.yaml
115 lines (115 loc) · 2.04 KB
/
model_config.yaml
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
train_ds:
manifest_filepath: /ws/manifests/raid/combined/train_manifest.json
sample_rate: 16000
labels: null
batch_size: 64
shuffle: true
time_length: 8
augmentor:
noise:
manifest_path: /ws/manifests/raid/musan/musan_music_noise_manifest_dur8.json
prob: 0.2
min_snr_db: 5
max_snr_db: 15
num_workers: 4
validation_ds:
manifest_filepath: /ws/manifests/raid/voxceleb/small_manifest.json
sample_rate: 16000
labels: null
batch_size: 64
shuffle: false
time_length: 8
num_workers: 1
preprocessor:
_target_: nemo.collections.asr.modules.AudioToMelSpectrogramPreprocessor
normalize: per_feature
window_size: 0.02
sample_rate: 16000
window_stride: 0.01
window: hann
features: 64
n_fft: 512
frame_splicing: 1
dither: 1.0e-05
stft_conv: false
encoder:
_target_: nemo.collections.asr.modules.ConvASREncoder
feat_in: 64
activation: relu
conv_mask: true
jasper:
- filters: 512
repeat: 1
kernel:
- 3
stride:
- 1
dilation:
- 1
dropout: 0.5
residual: true
separable: true
- filters: 512
repeat: 2
kernel:
- 7
stride:
- 1
dilation:
- 1
dropout: 0.5
residual: true
separable: true
- filters: 512
repeat: 2
kernel:
- 11
stride:
- 1
dilation:
- 1
dropout: 0.5
residual: true
separable: true
- filters: 512
repeat: 2
kernel:
- 15
stride:
- 1
dilation:
- 1
dropout: 0.5
residual: true
separable: true
- filters: 1500
repeat: 1
kernel:
- 1
stride:
- 1
dilation:
- 1
dropout: 0.0
residual: false
separable: true
decoder:
_target_: nemo.collections.asr.modules.SpeakerDecoder
feat_in: 1500
num_classes: 7205
pool_mode: xvector
emb_sizes: 256
angular: true
loss:
scale: 30
margin: 0.2
optim:
name: sgd
lr: 0.006
weight_decay: 0.001
sched:
name: CosineAnnealing
warmup_ratio: 0.1
min_lr: 0.0001
momentum: 0.9
target: nemo.collections.asr.models.label_models.EncDecSpeakerLabelModel