JorisCos commited on
Commit
9cd8979
1 Parent(s): 105bb97

Initial commit

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
+ - Libri3Mix
8
+ - sep_clean
9
+ license: CC BY-NC 4.0
10
+ inference: false
11
+ ---
12
+
13
+ ## Asteroid model `JorisCos/ConvTasNet_Libri3Mix_sepclean_8k`
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 `sep_clean` task of the Libri3Mix dataset.
19
+
20
+ Training config:
21
+ ```yml
22
+ data:
23
+ n_src: 3
24
+ sample_rate: 8000
25
+ segment: 3
26
+ task: sep_clean
27
+ train_dir: data/wav8k/min/train-360
28
+ valid_dir: data/wav8k/min/dev
29
+ filterbank:
30
+ kernel_size: 16
31
+ n_filters: 512
32
+ stride: 8
33
+ masknet:
34
+ bn_chan: 128
35
+ hid_chan: 512
36
+ mask_act: relu
37
+ n_blocks: 8
38
+ n_repeats: 3
39
+ n_src: 3
40
+ skip_chan: 128
41
+ optim:
42
+ lr: 0.001
43
+ optimizer: adam
44
+ weight_decay: 0.0
45
+ training:
46
+ batch_size: 24
47
+ early_stop: true
48
+ epochs: 200
49
+ half_lr: true
50
+ num_workers: 4
51
+ ```
52
+
53
+ Results :
54
+
55
+ On Libri3Mix min test set :
56
+ ```yaml
57
+ 'si_sdr': 8.581797049575108,
58
+ 'si_sdr_imp': 11.977037288467368,
59
+ 'sdr': 9.305885208641385,
60
+ 'sdr_imp': 12.3943409734845,
61
+ 'sir': 16.42030534048559,
62
+ 'sir_imp': 19.508759460400984,
63
+ 'sar': 10.641943911079238,
64
+ 'sar_imp': -56.4345187842095,
65
+ 'stoi': 0.8365148408724333,
66
+ 'stoi_imp': 0.24401766199806396
67
+ ```
68
+
69
+ License notice:
70
+
71
+ This work "ConvTasNet_Libri3Mix_sepclean_8k"
72
+ is a derivative of LibriSpeech ASR corpus by Vassil Panayotov,
73
+ used under CC BY 4.0. "ConvTasNet_Libri3Mix_sepclean_8k"
74
+ is licensed under Attribution-ShareAlike 3.0 Unported by Cosentino Joris.
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0939b6c934b48ea1111254a82b98484cd899b85c3c87eb63a13e4db244fdfec2
3
+ size 20593296