niobures commited on
Commit
c44513a
·
verified ·
1 Parent(s): 815e31f

U-Net (models)

Browse files
models/DCUNet_Libri1Mix_enhsingle_16k/.gitattributes ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.bin.* filter=lfs diff=lfs merge=lfs -text
2
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.h5 filter=lfs diff=lfs merge=lfs -text
5
+ *.tflite filter=lfs diff=lfs merge=lfs -text
6
+ *.tar.gz filter=lfs diff=lfs merge=lfs -text
7
+ *.ot filter=lfs diff=lfs merge=lfs -text
8
+ *.onnx filter=lfs diff=lfs merge=lfs -text
9
+ *.arrow filter=lfs diff=lfs merge=lfs -text
10
+ *.ftz filter=lfs diff=lfs merge=lfs -text
11
+ *.joblib filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.pb filter=lfs diff=lfs merge=lfs -text
15
+ *.pt filter=lfs diff=lfs merge=lfs -text
16
+ *.pth filter=lfs diff=lfs merge=lfs -text
models/DCUNet_Libri1Mix_enhsingle_16k/README.md ADDED
@@ -0,0 +1,72 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - asteroid
4
+ - audio
5
+ - DCUNet
6
+ - audio-to-audio
7
+ datasets:
8
+ - Libri1Mix
9
+ - enh_single
10
+ license: cc-by-sa-4.0
11
+ ---
12
+
13
+ ## Asteroid model `JorisCos/DCUNet_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_n_filters: 1024
32
+ stft_kernel_size: 1024
33
+ stft_stride: 256
34
+ masknet:
35
+ architecture: Large-DCUNet-20
36
+ fix_length_mode: pad
37
+ n_src: 1
38
+ optim:
39
+ lr: 0.001
40
+ optimizer: adam
41
+ weight_decay: 1.0e-05
42
+ training:
43
+ batch_size: 2
44
+ early_stop: true
45
+ epochs: 200
46
+ gradient_clipping: 5
47
+ half_lr: true
48
+ num_workers: 4
49
+ ```
50
+
51
+
52
+ Results:
53
+
54
+ On Libri1Mix min test set :
55
+ ```yml
56
+ si_sdr: 13.154035391645971
57
+ si_sdr_imp: 9.704254085786271
58
+ sdr: 13.568058873121435
59
+ sdr_imp: 10.065396073908367
60
+ sar: 13.568058873121435
61
+ sar_imp: 10.065396073908367
62
+ stoi: 0.9199373340235417
63
+ stoi_imp: 0.12401751048300132
64
+ ```
65
+
66
+
67
+ License notice:
68
+
69
+ This work "DCUNet_Libri1Mix_enhsignle_16k" is a derivative of [LibriSpeech ASR corpus](http://www.openslr.org/12) by Vassil Panayotov,
70
+ used under [CC BY 4.0](https://creativecommons.org/licenses/by/4.0/); of The WSJ0 Hipster Ambient Mixtures
71
+ 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).
72
+ "DCUNet_Libri1Mix_enhsignle_16k" is licensed under [Attribution-ShareAlike 3.0 Unported](https://creativecommons.org/licenses/by-sa/3.0/) by Joris Cosentino
models/DCUNet_Libri1Mix_enhsingle_16k/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:082a3f30c3f3a32f3c64ab7e6954b26fc5dd4ffa986bdd657049a9f3a93da7b6
3
+ size 39174139
models/DCUNet_Libri1Mix_enhsingle_16k/source.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ https://huggingface.co/JorisCos/DCUNet_Libri1Mix_enhsingle_16k
models/DCUnet/source.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ https://github.com/pheepa/DCUnet
models/DCUnet/weights.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8228d2dcb990a355a73ce0283bb301ef1d11ba20e8920e2f3bd175b038b2d817
3
+ size 11270124