ConvTasNet_Libri1Mix_enhsingle_8k

mpariente/ConvTasNet_Libri1Mix_enhsingle_8k

ConvTasNet_Libri1Mix_enhsingle_8k is a machine learning model from mpariente. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for asteroid. The card lists the license as cc-by-sa-4.0.

Downloads · 30 days

0% of all-time downloads

All-time downloads

24.6K

Public

Repo size

20.1 MB

Likes

Public

Hugging Face

Repo makeup

Click a slice to open those files.

.bin20.1 MB · 100%

At a glance

Library: asteroid
License: cc-by-sa-4.0
Access: Public
Created: Mar 2, 2022
Updated: Sep 23, 2021
SHA: 139ef2d9

Datasets

Model card

From the Hugging Face model README

Asteroid model

Imported from this Zenodo model page.

Description:

This model was trained by Brij Mohan using the Librimix/ConvTasNet recipe in Asteroid. It was trained on the enh_single task of the Libri3Mix dataset.

Training config:

data:
    n_src: 1
    sample_rate: 8000
    segment: 3
    task: enh_single
    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: 1
    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

Results:

si_sdr: 14.783675142685572
si_sdr_imp: 11.464625198953202
sdr: 15.497505907983102
sdr_imp: 12.07230150154914
sar: 15.497505907983102
sar_imp: 12.07230150154914
stoi: 0.9270030254700518
stoi_imp: 0.1320547197597893

License notice:

This work "ConvTasNet_Libri1Mix_enhsingle_8k" is a derivative of LibriSpeech ASR corpus by Vassil Panayotov, used under CC BY 4.0. "ConvTasNet_Libri1Mix_enhsingle_8k" is licensed under Attribution-ShareAlike 3.0 Unported by Manuel Pariente.

Library: asteroid
License: cc-by-sa-4.0
Created: Mar 2, 2022
Updated: Sep 23, 2021