Initial commit

README.md

@@ -0,0 +1,74 @@
+---
+tags:
+- asteroid
+- audio
+- ConvTasNet
+datasets:
+- Libri3Mix
+- sep_clean
+license: CC BY-NC 4.0
+inference: false
+---
+
+## Asteroid model `JorisCos/ConvTasNet_Libri3Mix_sepclean_8k`
+
+Description:
+
+This model was trained by Joris Cosentino using the librimix recipe in [Asteroid](https://github.com/asteroid-team/asteroid). 
+It was trained on the `sep_clean` task of the Libri3Mix dataset.
+
+Training config:
+```yml
+data:
+    n_src: 3
+    sample_rate: 8000
+    segment: 3
+    task: sep_clean
+    train_dir: data/wav8k/min/train-360
+    valid_dir: data/wav8k/min/dev
+filterbank:
+    kernel_size: 16
+    n_filters: 512
+    stride: 8
+masknet:
+    bn_chan: 128
+    hid_chan: 512
+    mask_act: relu
+    n_blocks: 8
+    n_repeats: 3
+    n_src: 3
+    skip_chan: 128
+optim:
+    lr: 0.001
+    optimizer: adam
+    weight_decay: 0.0
+training:
+  batch_size: 24
+  early_stop: true
+  epochs: 200
+  half_lr: true
+  num_workers: 4
+```
+
+Results :
+
+On Libri3Mix min test set :
+```yaml
+'si_sdr': 8.581797049575108,
+'si_sdr_imp': 11.977037288467368,
+'sdr': 9.305885208641385,
+'sdr_imp': 12.3943409734845,
+'sir': 16.42030534048559,
+'sir_imp': 19.508759460400984,
+'sar': 10.641943911079238,
+'sar_imp': -56.4345187842095,
+'stoi': 0.8365148408724333,
+'stoi_imp': 0.24401766199806396
+```
+
+License notice:
+
+This work "ConvTasNet_Libri3Mix_sepclean_8k" 
+is a derivative of LibriSpeech ASR corpus by Vassil Panayotov,
+used under CC BY 4.0. "ConvTasNet_Libri3Mix_sepclean_8k" 
+is licensed under Attribution-ShareAlike 3.0 Unported by Cosentino Joris.

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0939b6c934b48ea1111254a82b98484cd899b85c3c87eb63a13e4db244fdfec2
+size 20593296