Audio-to-Audio
speechbrain
English
speech-enhancement
PyTorch
pplantinga commited on
Commit
8f4ff7b
·
verified ·
1 Parent(s): caa614c

Upload 2 files

Browse files
Files changed (2) hide show
  1. hyperparams.yaml +46 -0
  2. score_model_ema.ckpt +3 -0
hyperparams.yaml ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ sample_rate: 16000
2
+ n_fft: 510
3
+ hop_length: 128
4
+ window_type: hann
5
+
6
+ transform_type: exponent
7
+ spec_factor: 0.15
8
+ spec_abs_exponent: 0.5
9
+
10
+ sampling:
11
+ sampler_type: pc
12
+ predictor: reverse_diffusion
13
+ corrector: ald
14
+ N: 30
15
+ corrector_steps: 1
16
+ snr: 0.5
17
+
18
+ modules:
19
+ score_model: !new:speechbrain.integrations.models.sgmse_plus.ScoreModel
20
+ backbone: ncsnpp_v2
21
+ sde: ouve
22
+ theta: 1.5
23
+ sigma_min: 0.05
24
+ sigma_max: 0.5
25
+ lr: 0.0001
26
+ ema_decay: 0.999
27
+ t_eps: 0.03
28
+ num_eval_files: 5
29
+ loss_type: score_matching
30
+ loss_weighting: sigma^2
31
+ network_scaling: 1/t
32
+ c_in: '1'
33
+ c_out: '1'
34
+ c_skip: '0'
35
+ sigma_data: 0.1
36
+ l1_weight: 0.001
37
+ pesq_weight: 0.0
38
+ N: 30
39
+ corrector_steps: 1
40
+ sampler_type: pc
41
+ snr: 0.5
42
+ sr: 16000
43
+
44
+ pretrainer: !new:speechbrain.utils.parameter_transfer.Pretrainer
45
+ loadables:
46
+ score_model_ema: !ref <modules[score_model]>
score_model_ema.ckpt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ca96321aca40cdca90c450d1450a5c7f343935e5b46ee34a1b575f9f774ccc3
3
+ size 262593305