File size: 442 Bytes
35a4689
 
 
 
 
79a2d3a
 
 
35a4689
 
 
 
 
 
 
 
 
79a2d3a
 
 
35a4689
79a2d3a
35a4689
79a2d3a
 
 
 
35a4689
79a2d3a
35a4689
79a2d3a
 
 
 
 
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
model_name: "rnnoise"

# spec
sample_rate: 8000
segment_size: 32000
nfft: 160
win_size: 160
hop_size: 80
win_type: hann

erb_bins: 32
min_freq_bins_for_erb: 2

# model
conv_size: 256
gru_size: 256

# data
max_snr_db: 20
min_snr_db: -10

# train
lr: 0.001
lr_scheduler: "CosineAnnealingLR"
lr_scheduler_kwargs:
  T_max: 250000
  eta_min: 0.0001

max_epochs: 100
clip_grad_norm: 10.0
seed: 1234

batch_size: 64
num_workers: 4
eval_steps: 15000