JorisCos commited on
Commit
101aeec
1 Parent(s): 0e6c27c

Initial commit

Browse files
Files changed (2) hide show
  1. README.md +73 -0
  2. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - asteroid
4
+ - audio
5
+ - ConvTasNet
6
+ datasets:
7
+ - Libri1Mix
8
+ - enh_single
9
+ license: CC BY-NC 4.0
10
+ inference: false
11
+ ---
12
+
13
+ ## Asteroid model `JorisCos/DCCRNet_Libri1Mix_enhsignle_16k`
14
+
15
+ Description:
16
+
17
+ This model was trained by Joris Cosentino using the librimix recipe in [Asteroid](https://github.com/asteroid-team/asteroid).
18
+ It was trained on the `enh_single` task of the Libri1Mix dataset.
19
+
20
+ Training config:
21
+
22
+ ```yml
23
+ data:
24
+ n_src: 1
25
+ sample_rate: 16000
26
+ segment: 3
27
+ task: enh_single
28
+ train_dir: data/wav16k/min/train-360
29
+ valid_dir: data/wav16k/min/dev
30
+ filterbank:
31
+ stft_kernel_size: 400
32
+ stft_n_filters: 512
33
+ stft_stride: 100
34
+ masknet:
35
+ architecture: DCCRN-CL
36
+ n_src: 1
37
+ optim:
38
+ lr: 0.001
39
+ optimizer: adam
40
+ weight_decay: 1.0e-05
41
+ training:
42
+ batch_size: 12
43
+ early_stop: true
44
+ epochs: 200
45
+ gradient_clipping: 5
46
+ half_lr: true
47
+ num_workers: 4
48
+ ```
49
+
50
+
51
+ Results:
52
+
53
+ On Libri1Mix min test set :
54
+ ```yml
55
+ si_sdr: 13.329767398333798,
56
+ si_sdr_imp: 9.879986092474098,
57
+ sdr: 13.87279932997016,
58
+ sdr_imp: 10.370136530757103,
59
+ sir: Infinity,
60
+ sir_imp: NaN,
61
+ sar: 13.87279932997016,
62
+ sar_imp: 10.370136530757103,
63
+ stoi: 0.9140907015623948,
64
+ stoi_imp: 0.11817087802185405
65
+ ```
66
+
67
+
68
+ License notice:
69
+
70
+ This work "DCCRNet_Libri1Mix_enhsignle_16k" is a derivative of [LibriSpeech ASR corpus](http://www.openslr.org/12) by Vassil Panayotov,
71
+ used under [CC BY 4.0](https://creativecommons.org/licenses/by/4.0/); of The WSJ0 Hipster Ambient Mixtures
72
+ dataset by [Whisper.ai](http://wham.whisper.ai/), used under [CC BY-NC 4.0](https://creativecommons.org/licenses/by-nc/4.0/) (Research only).
73
+ "DCCRNet_Libri1Mix_enhsignle_16k" is licensed under [Attribution-ShareAlike 3.0 Unported](https://creativecommons.org/licenses/by-sa/3.0/) by Joris Cosentino
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c1f519207604649cd90586e85adf2e93c4f4b26b216f0a71635013a90e64d939
3
+ size 16409109