HU Mathieu commited on
Commit
dd1f810
1 Parent(s): a3c3d06

Initial commit

Browse files

Files changed (2) hide show
  1. README.md +77 -0
  2. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1,77 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - asteroid
4
+ - audio
5
+ - ConvTasNet
6
+ datasets:
7
+ - libri1mix
8
+ - enh_single
9
+ license: cc-by-sa-3.0
10
+ inference: false
11
+ ---
12
+
13
+ ## Asteroid model `mhu-coder/ConvTasNet_Libri1Mix_enhsingle`
14
+ Imported from [Zenodo](https://zenodo.org/record/4301955#.X9cj98Jw0bY)
15
+
16
+ ### Description:
17
+ This model was trained by Mathieu Hu using the librimix/ConvTasNet recipe in
18
+ [Asteroid](https://github.com/asteroid-team/asteroid).
19
+ It was trained on the `enh_single` task of the Libri1Mix dataset.
20
+
21
+ ### Training config:
22
+ ```yaml
23
+ data:
24
+ n_src: 1
25
+ sample_rate: 16000
26
+ segment: 3
27
+ task: enh_single
28
+ train_dir: data/wav16k/min/train-100
29
+ valid_dir: data/wav16k/min/dev
30
+ filterbank:
31
+ kernel_size: 16
32
+ n_filters: 512
33
+ stride: 8
34
+ main_args:
35
+ exp_dir: exp/train_convtasnet_f34664b9
36
+ help: None
37
+ masknet:
38
+ bn_chan: 128
39
+ hid_chan: 512
40
+ mask_act: relu
41
+ n_blocks: 8
42
+ n_repeats: 3
43
+ n_src: 1
44
+ skip_chan: 128
45
+ optim:
46
+ lr: 0.001
47
+ optimizer: adam
48
+ weight_decay: 0.0
49
+ positional arguments:
50
+ training:
51
+ batch_size: 2
52
+ early_stop: True
53
+ epochs: 200
54
+ half_lr: True
55
+ num_workers: 4
56
+ ```
57
+
58
+ ### Results:
59
+ ```yaml
60
+ si_sdr: 13.938355526049932
61
+ si_sdr_imp: 10.488574220190232
62
+ sdr: 14.567380104207393
63
+ sdr_imp: 11.064717304994337
64
+ sir: inf
65
+ sir_imp: nan
66
+ sar: 14.567380104207393
67
+ sar_imp: 11.064717304994337
68
+ stoi: 0.9201010933251715
69
+ stoi_imp: 0.1241812697846321
70
+ ```
71
+
72
+ ### License notice:
73
+ This work "ConvTasNet_Libri1Mx_enhsingle" is a derivative of [CSR-I (WSJ0) Complete](https://catalog.ldc.upenn.edu/LDC93S6A)
74
+ by [LDC](https://www.ldc.upenn.edu/), used under [LDC User Agreement for
75
+ Non-Members](https://catalog.ldc.upenn.edu/license/ldc-non-members-agreement.pdf) (Research only).
76
+ "ConvTasNet_Libri1Mix_enhsingle" is licensed under [Attribution-ShareAlike 3.0 Unported](https://creativecommons.org/licenses/by-sa/3.0/)
77
+ by Mathieu Hu.
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18750591181961697bd84d1a99809aaecb6919f8bad1330b7c38ddc62aacbd27
3
+ size 20067856