mpariente commited on
Commit
5d14fc2
1 Parent(s): 6f12fcb

Import from Zenodo: v1

Browse files
Files changed (2) hide show
  1. README.md +74 -0
  2. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - asteroid
4
+ - audio
5
+ - ConvTasNet
6
+ datasets:
7
+ - LibriMix
8
+ - enh_single
9
+ license: cc-by-sa-3.0
10
+ inference: false
11
+ ---
12
+
13
+ ## Asteroid model
14
+ Imported from this Zenodo [model page](https://zenodo.org/record/3970768).
15
+
16
+ ## Description:
17
+ This model was trained by Brij Mohan using the Librimix/ConvTasNet recipe in Asteroid.
18
+ It was trained on the `enh_single` task of the Libri3Mix dataset.
19
+
20
+
21
+ ## Training config:
22
+ ```yaml
23
+ data:
24
+ n_src: 1
25
+ sample_rate: 8000
26
+ segment: 3
27
+ task: enh_single
28
+ train_dir: data/wav8k/min/train-360
29
+ valid_dir: data/wav8k/min/dev
30
+ filterbank:
31
+ kernel_size: 16
32
+ n_filters: 512
33
+ stride: 8
34
+ masknet:
35
+ bn_chan: 128
36
+ hid_chan: 512
37
+ mask_act: relu
38
+ n_blocks: 8
39
+ n_repeats: 3
40
+ n_src: 1
41
+ skip_chan: 128
42
+ optim:
43
+ lr: 0.001
44
+ optimizer: adam
45
+ weight_decay: 0.0
46
+ training:
47
+ batch_size: 24
48
+ early_stop: True
49
+ epochs: 200
50
+ half_lr: True
51
+ ```
52
+
53
+
54
+ ## Results:
55
+ ```yaml
56
+ si_sdr: 14.783675142685572
57
+ si_sdr_imp: 11.464625198953202
58
+ sdr: 15.497505907983102
59
+ sdr_imp: 12.07230150154914
60
+ sar: 15.497505907983102
61
+ sar_imp: 12.07230150154914
62
+ stoi: 0.9270030254700518
63
+ stoi_imp: 0.1320547197597893
64
+ ```
65
+
66
+
67
+ ## License notice:
68
+ This work "ConvTasNet_Libri1Mix_enhsingle_8k"
69
+ is a derivative of [LibriSpeech ASR corpus](http://www.openslr.org/12) by
70
+ [Vassil Panayotov](https://github.com/vdp),
71
+ used under [CC BY 4.0](https://creativecommons.org/licenses/by/4.0/).
72
+ "ConvTasNet_Libri1Mix_enhsingle_8k"
73
+ is licensed under [Attribution-ShareAlike 3.0 Unported](https://creativecommons.org/licenses/by-sa/3.0/)
74
+ by Manuel Pariente.
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c82d07cfb842778c26eeed222b313fcd9ae2776ce038a26f22dba0b700e597c
3
+ size 20063674